fix: clear stale model state on failed inference subprocess reload

2026-03-07 23:32:53 +00:00 · 2026-03-07 23:32:53 +00:00 · a7c34b42be
commit a7c34b42be
parent 4f766bbe25
1 changed files with 4 additions and 0 deletions
--- a/studio/backend/core/inference/orchestrator.py
+++ b/studio/backend/core/inference/orchestrator.py
@ -319,10 +319,14 @@ class InferenceOrchestrator:
            else:
                error = resp.get("error", "Failed to load model")
                self.loading_models.discard(model_name)
+                self.active_model_name = None
+                self.models.clear()
                raise Exception(error)

        except Exception:
            self.loading_models.discard(model_name)
+            self.active_model_name = None
+            self.models.clear()
            raise

    def unload_model(self, model_name: str) -> bool: