Update weight_utils.py

bbartels · web-flow · commit a03ea872d47b · 2025-07-21T22:03:46.000+01:00
diff --git a/vllm/model_executor/model_loader/weight_utils.py b/vllm/model_executor/model_loader/weight_utils.py
@@ -494,8 +494,7 @@ def runai_safetensors_weights_iterator(
             disable=not use_tqdm_on_load,
         )
 
-        for name, tensor in tensor_iter:
-            yield name, tensor
+        yield from tensor_iter
 
 def fastsafetensors_weights_iterator(
     hf_weights_files: list[str],

Original file line number	Diff line number	Diff line change
`@@ -494,8 +494,7 @@ def runai_safetensors_weights_iterator(`
`494`	`494`	`disable=not use_tqdm_on_load,`
`495`	`495`	`)`
`496`	`496`
`497`		`- for name, tensor in tensor_iter:`
`498`		`- yield name, tensor`
	`497`	`+ yield from tensor_iter`
`499`	`498`
`500`	`499`	`def fastsafetensors_weights_iterator(`
`501`	`500`	`hf_weights_files: list[str],`