From 846cc2cf2a4ebac1938035a89462de7e8c7e68a0 Mon Sep 17 00:00:00 2001 From: Roland Tannous Date: Tue, 10 Mar 2026 19:02:49 +0000 Subject: [PATCH] fix: always force-uncheck trainOnCompletions for pure audio models in dataset check Separate pure-audio from audio-VLM logic in runDatasetCheck so pure audio models are always forced to trainOnCompletions=false regardless of dataset type, while audio VLMs (gemma3n) only uncheck when the dataset is audio. --- .../src/features/training/stores/training-config-store.ts | 7 ++++++- 1 file changed, 6 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 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; } }