nodes/kijai-ComfyUI-ADMotionDirector
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix attempt

Browse Source
This commit is contained in:
kijai
2024-02-07 18:16:20 +02:00
parent e4b6df0e7a
commit d4ab6c4edf
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+10 -7
View File
@@ -812,14 +812,17 @@ class DiffusersLoaderForTraining:
def load_checkpoint(self, download_default, model=""):
with torch.inference_mode(False):
print(model)
if model != "":
model_path = model
if download_default and model != "stable-diffusion-v1-5":
from huggingface_hub import snapshot_download
download_to = os.path.join(folder_paths.models_dir,'diffusers')
snapshot_download(repo_id="runwayml/stable-diffusion-v1-5", ignore_patterns=["*.safetensors","*.ckpt", "*.pt", "*.png", "*non_ema*", "*fp16*"],
local_dir=f"{download_to}/stable-diffusion-v1-5", local_dir_use_symlinks=False)
model_path = "stable-diffusion-v1-5"
else:
if download_default and model_path != "stable-diffusion-v1-5":
from huggingface_hub import snapshot_download
download_to = os.path.join(folder_paths.models_dir,'diffusers')
snapshot_download(repo_id="runwayml/stable-diffusion-v1-5", ignore_patterns=["*.safetensors","*.ckpt", "*.pt", "*.png", "*non_ema*", "*fp16*"],
local_dir=f"{download_to}/stable-diffusion-v1-5", local_dir_use_symlinks=False)
model_path = model
for search_path in folder_paths.get_folder_paths("diffusers"):
if os.path.exists(search_path):