fix: FRAMEWORK vllm_bi150_r2r4 + image-verify CI (I8–I12)
All checks were successful
Docker Build / docker (push) Successful in 20m36s

This commit is contained in:
Victor Cao
2026-10-13 15:15:16 +08:00
parent fc15b61c81
commit 4c271270d4
12 changed files with 249 additions and 45 deletions

View File

@@ -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")