fix issues with 1.x SD models training

AIRI-AUBE · Mar 27, 2023 · ace896b · ace896b
1 parent ebdf607
commit ace896b
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/webui.py b/webui.py
@@ -719,11 +719,12 @@ def train():
 
             try:
                 print('Uploading SD Models...')
-                upload_s3files(
-                    sd_models_s3uri,
-                    os.path.join(sd_models_dir, db_model_name, f'{db_model_name}_*.yaml')
-                )
-                if db_save_safetensors:
+                if db_config.v2:
+                    upload_s3files(
+                        sd_models_s3uri,
+                        os.path.join(sd_models_dir, db_model_name, f'{db_model_name}_*.yaml')
+                    )
+                if db_config.save_safetensors:
                     upload_s3files(
                         sd_models_s3uri,
                         os.path.join(sd_models_dir, db_model_name, f'{db_model_name}_*.safetensors')