mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-24 17:35:39 +08:00
[BUG] Fix file parsing for load_format runai_streamer_sharded (#26324)
Signed-off-by: ahao-anyscale <ahao@anyscale.com>
This commit is contained in:
parent
c6873c4e6d
commit
4dbdf4a294
@ -121,7 +121,7 @@ class ShardedStateLoader(BaseModelLoader):
|
||||
|
||||
filepaths = []
|
||||
if is_s3(local_model_path):
|
||||
file_pattern = f"*{self.pattern.format(rank=rank, part=' * ')}"
|
||||
file_pattern = f"*{self.pattern.format(rank=rank, part='*')}"
|
||||
filepaths = s3_glob(path=local_model_path, allow_pattern=[file_pattern])
|
||||
else:
|
||||
filepaths = glob.glob(pattern)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user