xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-15 09:45:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Misc] add default value for file pattern arg (#21659)

Browse Source 
Signed-off-by: Andy Xie <andy.xning@gmail.com>
This commit is contained in:
Ning Xie 2025-07-27 13:14:51 +08:00 committed by GitHub
parent a8936e5193
commit 1cbf951ba2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/offline_inference/save_sharded_state.py
View File

@ -29,6 +29,7 @@ import shutil
from pathlib import Path from pathlib import Path
from vllm import LLM, EngineArgs from vllm import LLM, EngineArgs
from vllm.model_executor.model_loader import ShardedStateLoader
from vllm.utils import FlexibleArgumentParser from vllm.utils import FlexibleArgumentParser
@ -39,7 +40,10 @@ def parse_args():
"--output", "-o", required=True, type=str, help="path to output checkpoint" "--output", "-o", required=True, type=str, help="path to output checkpoint"
) )
parser.add_argument( parser.add_argument(
"--file-pattern", type=str, help="string pattern of saved filenames" "--file-pattern",
type=str,
default=ShardedStateLoader.DEFAULT_PATTERN,
help="string pattern of saved filenames",
) )
parser.add_argument( parser.add_argument(
"--max-file-size", "--max-file-size",