From 294a3d3e478e619188040a5e0f1601ca5fb5f07a Mon Sep 17 00:00:00 2001
From: Manan17 <shahmanan170602@gmail.com>
Date: Tue, 10 Mar 2026 21:33:13 +0000
Subject: [PATCH] fix: reset isEmbeddingModel in error fallback paths to
 prevent stale state

---
 .../src/features/training/stores/training-config-store.ts     | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/studio/frontend/src/features/training/stores/training-config-store.ts b/studio/frontend/src/features/training/stores/training-config-store.ts
index 56460137a9..cbaf9f375c 100644
--- a/studio/frontend/src/features/training/stores/training-config-store.ts
+++ b/studio/frontend/src/features/training/stores/training-config-store.ts
@@ -165,6 +165,7 @@ export const useTrainingConfigStore = create<TrainingConfigStore>()(
 
             set({
               isLoadingModelDefaults: false,
+              isEmbeddingModel: false,
               isAudioModel: false,
               modelDefaultsError:
                 error instanceof Error
@@ -179,13 +180,14 @@ export const useTrainingConfigStore = create<TrainingConfigStore>()(
                 set({
                   modelType: isVision ? "vision" : "text",
                   isVisionModel: isVision,
+                  isEmbeddingModel: false,
                   isAudioModel: false,
                   isCheckingVision: false,
                 });
               })
               .catch(() => {
                 if (get().selectedModel !== modelName) return;
-                set({ isCheckingVision: false, isAudioModel: false });
+                set({ isCheckingVision: false, isEmbeddingModel: false, isAudioModel: false });
               });
           });
       };