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

model stuff

Browse Source
This commit is contained in:
firestar5683
2026-08-11 09:30:46 -05:00
parent d7ef213792
commit acf32365be
5 changed files with 199 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/model_compiler.py
+7 -5
View File
@@ -552,14 +552,16 @@ def compile_driving(
) -> Path:
model_type, source_args = driving_compile_args(files, input_format)
output_path = output_dir / f"{model_key}_driving_tinygrad.pkl"
# A rebuild queue may compile several models into the same directory. Only
# replace the selected model; deleting every driving artifact here loses
# models that were successfully compiled earlier in the queue.
removed = remove_paths(sorted({
output_path,
*multipart_output_paths(output_path, output_dir),
*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"),
*output_dir.glob(f"{model_key}_driving_*_tinygrad.pkl"),
*output_dir.glob(f"{model_key}_driving_*_tinygrad.pkl.p[0-9][0-9]"),
*output_dir.glob(f"{model_key}_driving_*_tinygrad.pkl.sha256"),
*output_dir.glob(f"{model_key}_driving_*_metadata.pkl"),
}))
if removed:
print(f" cleared {removed} existing driving output entries")