diff --git a/cspnodes.py b/cspnodes.py index 049d632..f903634 100644 --- a/cspnodes.py +++ b/cspnodes.py @@ -1,5 +1,8 @@ import os import torch +import huggingface_hub +if not hasattr(huggingface_hub, "cached_download"): + huggingface_hub.cached_download = huggingface_hub.hf_hub_download from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler from PIL import Image, ImageOps import numpy as np