We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ddfdf47 commit 75beba2Copy full SHA for 75beba2
vllm/model_executor/weight_utils.py
@@ -39,7 +39,10 @@ def hf_model_weights_iterator(
39
else:
40
hf_folder = model_name_or_path
41
42
- hf_bin_files = glob.glob(os.path.join(hf_folder, "*.bin"))
+ hf_bin_files = [
43
+ x for x in glob.glob(os.path.join(hf_folder, "*.bin"))
44
+ if not x.endswith("training_args.bin")
45
+ ]
46
47
if use_np_cache:
48
# Convert the model weights from torch tensors to numpy arrays for
0 commit comments