Unverified Commit 1cbf951b authored by Ning Xie's avatar Ning Xie Committed by GitHub
Browse files

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


Signed-off-by: default avatarAndy Xie <andy.xning@gmail.com>
parent a8936e51
...@@ -29,6 +29,7 @@ import shutil ...@@ -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(): ...@@ -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",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment