diff --git a/nodes.py b/nodes.py index e4b01fc..ae32723 100644 --- a/nodes.py +++ b/nodes.py @@ -31,7 +31,7 @@ KDPM2DiscreteScheduler, ) -from huggingface_hub import snapshot_download +from huggingface_hub import hf_hub_download, model_info from plyfile import PlyData import trimesh @@ -1537,7 +1537,7 @@ def load_diffusers_pipe(self, diffusers_pipeline_name, repo_id, custom_pipeline, # resume download pretrained checkpoint ckpt_download_dir = os.path.join(CKPT_DIFFUSERS_PATH, repo_id) - snapshot_download(repo_id=repo_id, local_dir=ckpt_download_dir, force_download=force_download, repo_type="model", ignore_patterns=HF_DOWNLOAD_IGNORE) + hf_hub_download(repo_id=repo_id, local_dir=ckpt_download_dir, force_download=force_download, repo_type="model", ignore_patterns=HF_DOWNLOAD_IGNORE) diffusers_pipeline_class = DIFFUSERS_PIPE_DICT[diffusers_pipeline_name]