[RL] add --skip-warmup (#7416)

This commit is contained in:
Zilin Zhu
2025-07-03 09:50:43 +08:00
committed by GitHub
parent b116b21a46
commit 09e699bba4
2 changed files with 24 additions and 5 deletions

View File

@@ -46,6 +46,7 @@ class ServerArgs:
tokenizer_path: Optional[str] = None
tokenizer_mode: str = "auto"
skip_tokenizer_init: bool = False
skip_server_warmup: bool = False
load_format: str = "auto"
model_loader_extra_config: str = "{}"
trust_remote_code: bool = False
@@ -597,6 +598,11 @@ class ServerArgs:
action="store_true",
help="If set, skip init tokenizer and pass input_ids in generate request.",
)
parser.add_argument(
"--skip-server-warmup",
action="store_true",
help="If set, skip warmup.",
)
parser.add_argument(
"--load-format",
type=str,