diff --git a/studio/backend/core/inference/orchestrator.py b/studio/backend/core/inference/orchestrator.py index a6c297c046..5db7e56357 100644 --- 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: