cleanup

AIRI-AUBE · Mar 27, 2023 · 5673c52 · 5673c52
1 parent 35e5fc0
commit 5673c52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/webui.py b/webui.py
@@ -747,7 +747,7 @@ def train():
                 #automatic tar latest checkpoint and upload to s3 by zheng on 2023.03.22
                 os.makedirs(os.path.dirname("/opt/ml/model/"), exist_ok=True)
                 os.makedirs(os.path.dirname("/opt/ml/model/Stable-diffusion/"), exist_ok=True)
-                os.makedirs(os.path.dirname("/opt/ml/model/Lora/"), exist_ok=True)
+                os.makedirs(os.path.dirname("/opt/ml/model/ControlNet/"), exist_ok=True)
                 train_steps=int(db_config.revision)
                 model_file_basename = f'{db_model_name}_{train_steps}_lora' if db_config.use_lora else f'{db_model_name}_{train_steps}'
                 f1=os.path.join(sd_models_dir, db_model_name, f'{model_file_basename}.yaml')