xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-07-30 14:21:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Fix] The zip function in Python 3.9 does not have the strict argument (#19549)

Browse Source 
Signed-off-by: 汪志鹏 <wangzhipeng628@gmail.com>
This commit is contained in:
汪志鹏 2025-06-13 14:57:48 +08:00 committed by GitHub
parent ace5cdaff0
commit cefdb9962d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
vllm/v1/worker/gpu_model_runner.py
View File

@ -467,10 +467,8 @@ class GPUModelRunner(LoRAModelRunnerMixin):
# Update the block IDs. # Update the block IDs.
if not req_data.resumed_from_preemption: if not req_data.resumed_from_preemption:
# Append the new blocks to the existing block IDs. # Append the new blocks to the existing block IDs.
for block_ids, new_block_ids in zip( # type: ignore[call-overload] for block_ids, new_block_ids in zip(req_state.block_ids,
req_state.block_ids, req_data.new_block_ids):
req_data.new_block_ids,
strict=True):
block_ids.extend(new_block_ids) block_ids.extend(new_block_ids)
else: else:
# The request is resumed from preemption. # The request is resumed from preemption.

6
vllm/v1/worker/tpu_model_runner.py
View File

@ -413,10 +413,8 @@ class TPUModelRunner(LoRAModelRunnerMixin):
req_state.num_computed_tokens = req_data.num_computed_tokens req_state.num_computed_tokens = req_data.num_computed_tokens
if not req_data.resumed_from_preemption: if not req_data.resumed_from_preemption:
# Append the new blocks to the existing block IDs. # Append the new blocks to the existing block IDs.
for block_ids, new_block_ids in zip( # type: ignore[call-overload] for block_ids, new_block_ids in zip(req_state.block_ids,
req_state.block_ids, req_data.new_block_ids):
req_data.new_block_ids,
strict=True):
block_ids.extend(new_block_ids) block_ids.extend(new_block_ids)
else: else:
# The request is resumed from preemption. # The request is resumed from preemption.