From cfad8ec36d4bf3476d79287e840fa650cf76ffd1 Mon Sep 17 00:00:00 2001 From: Roland Tannous Date: Sat, 7 Mar 2026 23:29:34 +0000 Subject: [PATCH] fix: reset checkpoint metadata on failed export checkpoint reload --- studio/backend/core/export/orchestrator.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/studio/backend/core/export/orchestrator.py b/studio/backend/core/export/orchestrator.py index 47bb73daeb..ec18d3533e 100644 --- a/studio/backend/core/export/orchestrator.py +++ b/studio/backend/core/export/orchestrator.py @@ -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(