[BugFix] Fix a bug in loading safetensors (vllm-project#1732)

hongxiayang · Nov 20, 2023 · fb6865a · fb6865a
1 parent dfaf303
commit fb6865a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm/model_executor/weight_utils.py b/vllm/model_executor/weight_utils.py
@@ -240,7 +240,7 @@ def hf_model_weights_iterator(
     elif use_safetensors:
         for st_file in hf_weights_files:
             with safe_open(st_file, framework="pt") as f:
-                for name in f:
+                for name in f.keys():  # noqa: SIM118
                     param = f.get_tensor(name)
                     yield name, param
     else: