nodes/MinusZoneAI-ComfyUI-TrainTools-MZ
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

'update'

Browse Source
This commit is contained in:
wailovet
2024-06-29 10:17:23 +08:00
parent 32b3c6514a
commit c4d76a1dbb
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mz_train_tools_core_HYDiT.py
+3 -2
View File
@@ -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)