This commit is contained in:
wassname
2026-04-27 11:24:19 +08:00
parent 24ba8deb02
commit a342801807
5 changed files with 18 additions and 35 deletions
+1 -1
View File
@@ -428,7 +428,7 @@ def check_probe_reload(
del loaded_model
gc.collect()
torch.cuda.empty_cache()
return {"reload_err": reload_err, "saved_tensors": len(saved["state"])}
return {"reload_err": reload_err, "saved_tensors": len(saved_sd)}
def print_final_report(row: dict[str, Any], result_path: Path, mode: str) -> None: