Update webui.py

AIRI-AUBE · Mar 22, 2023 · ad48c63 · ad48c63
1 parent ef8405b
commit ad48c63
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/webui.py b/webui.py
@@ -742,7 +742,8 @@ def train():
                     shutil.copy(f2,"/opt/ml/model/")
                 f3=os.path.join(sd_models_path, f'{sd_models_path}/{db_model_name}_{train_steps}.safetensors')
                 if os.path.exists(f3):
-                    shutil.copy(f3,"/opt/ml/model/")            except Exception as e:
+                    shutil.copy(f3,"/opt/ml/model/")
+            except Exception as e:
                 traceback.print_exc()
                 print(e)
             opts.data = default_options