fix

vms/ui/app_ui.py

@@ -234,7 +234,7 @@ class AppUI:
                     self.project_tabs["train_tab"].components["num_gpus"],
                     self.project_tabs["train_tab"].components["precomputation_items"],
                     self.project_tabs["train_tab"].components["lr_warmup_steps"],
-                    self.project_tabs["train_tab"].components["auto_resume_checkbox"]
+                    self.project_tabs["train_tab"].components["auto_resume"]
                 ]
             )
         

vms/ui/project/services/training.py

@@ -1078,7 +1078,8 @@ class TrainingService:
                             "learning_rate": ui_state.get("learning_rate", DEFAULT_LEARNING_RATE),
                             "save_iterations": ui_state.get("save_iterations", DEFAULT_SAVE_CHECKPOINT_EVERY_N_STEPS),
                             "preset_name": ui_state.get("training_preset", list(TRAINING_PRESETS.keys())[0]),
-                            "repo_id": ""  # Default empty repo ID
+                            "repo_id": ""  # Default empty repo ID,
+                            "auto_resume": ui_state.get("auto_resume", DEFAULT_AUTO_RESUME)
                         }
                     }
                     logger.info("Created default session from UI state for recovery")
@@ -1150,7 +1151,7 @@ class TrainingService:
                 "learning_rate": params.get('learning_rate', DEFAULT_LEARNING_RATE),
                 "save_iterations": params.get('save_iterations', DEFAULT_SAVE_CHECKPOINT_EVERY_N_STEPS),
                 "training_preset": params.get('preset_name', list(TRAINING_PRESETS.keys())[0]),
-                "auto_resume_checkbox": params.get("auto_resume", DEFAULT_AUTO_RESUME)
+                "auto_resume": params.get("auto_resume", DEFAULT_AUTO_RESUME)
             })
             
             # Check if we should auto-recover (immediate restart)

vms/ui/project/tabs/train_tab.py

@@ -233,7 +233,7 @@ class TrainTab(BaseTab):
                                     )
 
                                 with gr.Row():
-                                    self.components["auto_resume_checkbox"] = gr.Checkbox(
+                                    self.components["auto_resume"] = gr.Checkbox(
                                         label="Automatically continue training in case of server reboot.",
                                         value=DEFAULT_AUTO_RESUME,
                                         info="When enabled, training will automatically resume from the latest checkpoint after app restart"
@@ -389,9 +389,9 @@ class TrainTab(BaseTab):
             ]
         )
 
-        self.components["auto_resume_checkbox"].change(
+        self.components["auto_resume"].change(
             fn=lambda v: self.app.update_ui_state(auto_resume=v),
-            inputs=[self.components["auto_resume_checkbox"]],
+            inputs=[self.components["auto_resume"]],
             outputs=[]
         )