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 3a182b73ec..30d03e7fc5 100644 --- a/studio/frontend/src/features/training/stores/training-config-store.ts +++ b/studio/frontend/src/features/training/stores/training-config-store.ts @@ -213,7 +213,12 @@ export const useTrainingConfigStore = create()( if (isVisionModel && isImage) { updates.trainOnCompletions = false; } - if (isAudioModel && isAudio) { + // Pure audio model → always uncheck regardless of dataset. + if (isAudioModel && !isVisionModel) { + updates.trainOnCompletions = false; + } + // Audio-capable vision model (e.g. gemma3n) + audio dataset → uncheck. + if (isAudioModel && isVisionModel && isAudio) { updates.trainOnCompletions = false; } }