mawei/StarPilot
1
0
Fork 0
mirror of https://github.com/firestar5683/StarPilot.git synced 2026-08-31 21:23:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

v16

Browse Source
This commit is contained in:
firestar5683
2026-08-27 15:11:41 -05:00
parent 3e176f3884
commit b2602bc4e7
2 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/model_compiler.py
+9 -3
View File
@@ -48,6 +48,7 @@ MODEL_CONTEXT_FREQ = 5
REPOSITORY_FILE_LIMIT = 100_000_000
DEFAULT_MULTIPART_SIZE = 95 * 1024 * 1024
USBGPU_PROBE_ATTEMPTS = 10
DEFAULT_SUPERCOMBO_BEHAVIOR_VERSION = "v16"
def build_compile_env(*, supercombo: bool = False) -> dict[str, str]:
@@ -280,6 +281,13 @@ def infer_model_version(model_key: str, explicit_version: str | None) -> str:
return ""
def resolve_behavior_version(model_key: str, explicit_version: str | None, input_format: str) -> str:
version = infer_model_version(model_key, explicit_version)
if not version and input_format == "supercombo":
return DEFAULT_SUPERCOMBO_BEHAVIOR_VERSION
return version
def select_input_format(requested: str, files: dict[str, Path]) -> str:
if requested == "supercombo":
if "driving_supercombo" not in files:
@@ -661,9 +669,7 @@ def main() -> int:
source_path = Path(source)
validate_onnx_source(source_path)
print(f" source {option.removeprefix('--')}: {source_path} ({source_path.stat().st_size} bytes)")
version = infer_model_version(model_key, args.version)
if not version and input_format == "supercombo":
version = "v15"
version = resolve_behavior_version(model_key, args.version, input_format)
version_label = version or "unspecified behavior"
will_install = model_key.startswith("local-") and not args.no_install
target = f"{args.output_dir}" + (f" -> {MODELS_PATH} (auto-install)" if will_install else "")