Spaces:

multimodalart
/

dreambooth-training

Running

multimodalart HF Staff commited on Dec 22, 2022

Commit

a2f9429

1 Parent(s): 21f25c7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -297,8 +297,8 @@ def train(*inputs):
     torch.cuda.empty_cache()
     if(which_model == "v1-5"):
         print("Adding Safety Checker to the model...")
-        shutil.copytree(f"{safety_checker}/feature_extractor", "output_model/feature_extractor")
-        shutil.copytree(f"{safety_checker}/safety_checker", "output_model/safety_checker")
         shutil.copy(f"model_index.json", "output_model/model_index.json")
     if(not remove_attribution_after):

     torch.cuda.empty_cache()
     if(which_model == "v1-5"):
         print("Adding Safety Checker to the model...")
+        shutil.copytree(f"{safety_checker}/feature_extractor", "output_model/feature_extractor", dirs_exist_ok=True)
+        shutil.copytree(f"{safety_checker}/safety_checker", "output_model/safety_checker", dirs_exist_ok=True)
         shutil.copy(f"model_index.json", "output_model/model_index.json")
     if(not remove_attribution_after):