From 5673c52ff3e02ee15588dd9559e4c5f76020e015 Mon Sep 17 00:00:00 2001 From: xieyongliang Date: Mon, 27 Mar 2023 08:23:11 +0800 Subject: [PATCH] cleanup --- webui.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/webui.py b/webui.py index 54b36a932ff..e117d4ede84 100644 --- 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')