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

Commit

Permalink
[Minor] Fix the path typo in loader.py: save_sharded_states.py -> sav…
Browse files Browse the repository at this point in the history
…e_sharded_state.py (vllm-project#5151)

Signed-off-by: Ye Cao <caoye.cao@alibaba-inc.com>
  • Loading branch information
dashanji authored and robertgshaw2-neuralmagic committed Jun 11, 2024
1 parent db09745 commit 46b6b26
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 @@ -407,7 +407,7 @@ class ShardedStateLoader(BaseModelLoader):
Model loader that directly loads each worker's model state dict, which
enables a fast load path for large tensor-parallel models where each worker
only needs to read its own shard rather than the entire checkpoint. See
`examples/save_sharded_states.py` for creating a sharded checkpoint.
`examples/save_sharded_state.py` for creating a sharded checkpoint.
"""

DEFAULT_PATTERN = "model-rank-{rank}-part-{part}.safetensors"
Expand Down

0 comments on commit 46b6b26

Please sign in to comment.