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 dbf5b3a817..260625db0d 100644
--- a/studio/frontend/src/features/training/stores/training-config-store.ts
+++ b/studio/frontend/src/features/training/stores/training-config-store.ts
@@ -46,11 +46,9 @@ let _modelConfigController: AbortController | null = null;
 const NON_PERSISTED_STATE_KEYS: ReadonlySet<keyof TrainingConfigState> = new Set([
   "modelType",
   "isCheckingVision",
-  "isVisionModel",
   "isLoadingModelDefaults",
   "modelDefaultsError",
   "isCheckingDataset",
-  "isDatasetMultimodal",
 ]);
 
 function partializePersistedState(