fix: FRAMEWORK vllm_bi150_r2r4 + image-verify CI (I8–I12)
All checks were successful
Docker Build / docker (push) Successful in 20m36s
All checks were successful
Docker Build / docker (push) Successful in 20m36s
This commit is contained in:
@@ -12,6 +12,7 @@ import sys
|
||||
from pathlib import Path
|
||||
|
||||
FORBIDDEN_FRAMEWORK = "vllm_fix_tokenizer"
|
||||
DEFAULT_FRAMEWORK = "vllm_bi150_r2r4"
|
||||
TOKENIZER_PATCH_KEY = "modelhub_tokenizer_patch"
|
||||
|
||||
|
||||
@@ -26,7 +27,7 @@ def engine_root() -> Path:
|
||||
def assert_framework(framework: str) -> None:
|
||||
if not framework or framework == FORBIDDEN_FRAMEWORK:
|
||||
raise PatchError(
|
||||
"FRAMEWORK 必须是 vllm_0_17_0_corex_4_4_0,"
|
||||
f"FRAMEWORK 必须是 {DEFAULT_FRAMEWORK},"
|
||||
"不能使用 vllm_fix_tokenizer"
|
||||
)
|
||||
|
||||
@@ -157,7 +158,7 @@ def main(argv: list[str] | None = None) -> int:
|
||||
root = engine_root()
|
||||
parser = argparse.ArgumentParser(description="应用 R2/R4 模型目录修补")
|
||||
parser.add_argument("--model-dir", required=True, type=Path)
|
||||
parser.add_argument("--framework", default="vllm_0_17_0_corex_4_4_0")
|
||||
parser.add_argument("--framework", default=DEFAULT_FRAMEWORK)
|
||||
parser.add_argument("--unknown-policy", default="fail", choices=("fail", "skip"))
|
||||
parser.add_argument("--force-chat-template", action="store_true")
|
||||
parser.add_argument("--tokenizer-patch", action="store_true")
|
||||
|
||||
Reference in New Issue
Block a user