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

More modeld cleanup

Browse Source
This commit is contained in:
firestar5683
2026-06-28 15:03:09 -05:00
parent f7418649f8
commit 40b5468615
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/model_compiler.py
+10 -5
View File
@@ -345,14 +345,17 @@ def split_oversized_artifact(
def compile_driving(model_key: str, files: dict[str, Path], input_format: str, version: str, output_dir: Path) -> Path:
model_type, source_args = driving_compile_args(files, input_format)
output_path = output_dir / f"{model_key}_driving_tinygrad.pkl"
removed = remove_paths([
removed = remove_paths(sorted({
output_path,
*multipart_output_paths(output_path, output_dir),
*output_dir.glob(f"{model_key}_driving_*_tinygrad.pkl"),
*output_dir.glob(f"{model_key}_driving_*_metadata.pkl"),
])
*output_dir.glob("*_driving_tinygrad.pkl"),
*output_dir.glob("*_driving_tinygrad.pkl.p[0-9][0-9]"),
*output_dir.glob("*_driving_tinygrad.pkl.sha256"),
*output_dir.glob("*_driving_*_tinygrad.pkl"),
*output_dir.glob("*_driving_*_metadata.pkl"),
}))
if removed:
print(f" cleared {removed} existing output entries for {model_key}")
print(f" cleared {removed} existing driving output entries")
frame_skip = MODEL_RUN_FREQ // MODEL_CONTEXT_FREQ
command = [
@@ -476,6 +479,8 @@ def main() -> int:
print(" artifact exceeds 100 MiB; created repository-safe multipart files:")
for multipart_output in multipart_outputs:
print(f" {multipart_output.name} ({multipart_output.stat().st_size} bytes)")
output.unlink()
print(f" removed oversized source artifact {output.name}")
print("Done.")
return 0