From c4d76a1dbb838bfe8f486afa7b4b739e493c5e9a Mon Sep 17 00:00:00 2001 From: wailovet Date: Sat, 29 Jun 2024 10:17:23 +0800 Subject: [PATCH] 'update' --- mz_train_tools_core_HYDiT.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/mz_train_tools_core_HYDiT.py b/mz_train_tools_core_HYDiT.py index b30d9b8..b304d98 100644 --- a/mz_train_tools_core_HYDiT.py +++ b/mz_train_tools_core_HYDiT.py @@ -215,7 +215,8 @@ def check_model_auto_download(args): download_file = "HunyuanDiT/t2i/model/pytorch_model_module.pt" download_fullpath = os.path.join( hunyuan_base_path, download_file) - os.makedirs(os.path.dirname(download_fullpath), exist_ok=True) + if not os.path.exists(download_fullpath): + os.makedirs(os.path.dirname(download_fullpath), exist_ok=True) if os.path.exists(download_fullpath): args["unet_path"] = download_fullpath @@ -237,7 +238,7 @@ def check_model_auto_download(args): for download_file in download_files: download_fullpath = os.path.join( - hunyuan_base_path, download_file) + hunyuan_base_path, download_file) if not os.path.exists(download_fullpath): os.makedirs(os.path.dirname(download_fullpath), exist_ok=True)