Skip to content

Commit 17b4045

Browse files
authored
Update weight_utils.py
1 parent ad67e10 commit 17b4045

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

vllm/model_executor/model_loader/weight_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -475,6 +475,7 @@ def safetensors_weights_iterator(
475475
param = f.get_tensor(name)
476476
yield name, param
477477

478+
478479
def runai_safetensors_weights_iterator(
479480
hf_weights_files: list[str],
480481
use_tqdm_on_load: bool,
@@ -496,6 +497,7 @@ def runai_safetensors_weights_iterator(
496497

497498
yield from tensor_iter
498499

500+
499501
def fastsafetensors_weights_iterator(
500502
hf_weights_files: list[str],
501503
use_tqdm_on_load: bool,

0 commit comments

Comments
 (0)