Skip to content
This repository has been archived by the owner on Oct 11, 2024. It is now read-only.

Commit

Permalink
[Bugfix] Ensure download_weights_from_hf(..) inside loader is using t…
Browse files Browse the repository at this point in the history
…he revision parameter (vllm-project#4217)
  • Loading branch information
alexm-redhat authored and robertgshaw2-redhat committed Apr 26, 2024
1 parent 7e236d1 commit 0065ecb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion vllm/model_executor/model_loader/loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -172,7 +172,7 @@ def _prepare_weights(self, model_name_or_path: str,
if not is_local:
hf_folder = download_weights_from_hf(model_name_or_path,
self.load_config.download_dir,
allow_patterns)
allow_patterns, revision)
else:
hf_folder = model_name_or_path

Expand Down

0 comments on commit 0065ecb

Please sign in to comment.