Handle non-critical errors, outputs.

Author: d8ahazard

dreambooth/dataclasses/db_config.py

@@ -202,12 +202,21 @@ def save(self, backup=False):
 
         if backup:
             backup_dir = os.path.join(models_path, "backups")
-            if not os.path.exists(backup_dir):
-                os.makedirs(backup_dir)
-            config_file = os.path.join(models_path, "backups", f"db_config_{self.revision}.json")
+            try:
+                if not os.path.exists(backup_dir):
+                    os.makedirs(backup_dir)
+                config_file = os.path.join(models_path, "backups", f"db_config_{self.revision}.json")
+            except Exception as e:
+                # Read-only or permission issues: skip backup silently but continue without failing
+                logger.warning(f"Skipping config backup (read-only or permission issue): {e}")
+                config_file = os.path.join(models_path, "db_config.json")
 
-        with open(config_file, "w") as outfile:
-            json.dump(self.__dict__, outfile, indent=4)
+        try:
+            with open(config_file, "w") as outfile:
+                json.dump(self.__dict__, outfile, indent=4)
+        except Exception as e:
+            # Do not break training/UI on read-only filesystems
+            logger.warning(f"Skipping config save to '{config_file}' (read-only or permission issue): {e}")
 
     def load_params(self, params_dict):
         sched_swap = False

scripts/main.py

@@ -1869,7 +1869,7 @@ def class_gen_method_changed(method):
         )
 
         db_generate_sample.click(
-            fn=wrap_gpu_call(generate_samples),
+            fn=wrap_gpu_call(generate_samples, extra_outputs=[[], ""]),
             _js="db_start_sample",
             inputs=[
                 db_model_name,
@@ -1896,7 +1896,7 @@ def class_gen_method_changed(method):
 
         db_generate_checkpoint.click(
             _js="db_start_checkpoint",
-            fn=wrap_gpu_call(ui_gen_ckpt),
+            fn=wrap_gpu_call(ui_gen_ckpt, extra_outputs=[]),
             inputs=[db_model_name],
             outputs=[db_status],
         )
@@ -1912,7 +1912,7 @@ def set_gen_sample():
         db_train_sample.click(fn=set_gen_sample, inputs=[], outputs=[])
 
         db_create_model.click(
-            fn=wrap_gpu_call(create_model),
+            fn=wrap_gpu_call(create_model, extra_outputs=[None, None, None, None, None, None, None, None, None]),
             _js="db_start_create",
             inputs=[
                 db_new_model_name,
@@ -1940,15 +1940,15 @@ def set_gen_sample():
         )
 
         db_train_model.click(
-            fn=wrap_gpu_call(start_training),
+            fn=wrap_gpu_call(start_training, extra_outputs=[gr.update(visible=True), 0, 0, []]),
             _js="db_start_train",
             inputs=[db_model_name, db_class_gen_method],
             outputs=[db_lora_model_name, db_revision, db_epochs, db_gallery, db_status],
         )
 
         db_generate_classes.click(
             _js="db_start_classes",
-            fn=wrap_gpu_call(ui_classifiers),
+            fn=wrap_gpu_call(ui_classifiers, extra_outputs=[[]]),
             inputs=[db_model_name, db_class_gen_method],
             outputs=[db_gallery, db_status],
         )