github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-07 12:01:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix CI bugs

Browse Source
This commit is contained in:
yuehuayingxueluo
2024-01-09 15:18:28 +08:00
committed by FrankLeeeee
parent 2a73e828eb
commit fab294c7f4
5 changed files with 21 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_infer/test_request_handler.py
View File

@@ -57,7 +57,7 @@ def check_request_handler():
block_size=16,
eos_token_id=0,
sample_params=None,
block_table=torch.tensor([0, 0]),
block_table=torch.tensor([-1, -1]),
)
request_handler.add_sequence(seq1)
# the priority should be 1