ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: reset checkpoint metadata on failed export checkpoint reload

Browse source
This commit is contained in:
Roland Tannous 2026-03-07 23:29:34 +00:00
commit cfad8ec36d
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
studio/backend/core/export/orchestrator.py
View file

@ -240,6 +240,9 @@ class ExportOrchestrator:
resp = self._wait_response("loaded", timeout=300)
except RuntimeError as exc:
self._shutdown_subprocess(timeout=5)
self.current_checkpoint = None
self.is_vision = False
self.is_peft = False
return False, str(exc)
if resp.get("success"):
@ -251,6 +254,9 @@ class ExportOrchestrator:
else:
error = resp.get("message", "Failed to load checkpoint")
logger.error("Failed to load checkpoint: %s", error)
self.current_checkpoint = None
self.is_vision = False
self.is_peft = False
return False, error
def export_merged_model(