Fixed constants.py not using hugging face hub environment variable (h…

…uggingface#5222)
jamarju · Oct 2, 2023 · 907fd91 · 907fd91
1 parent 0c7cb9a
commit 907fd91
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/utils/constants.py b/src/diffusers/utils/constants.py
@@ -25,7 +25,7 @@
 ONNX_WEIGHTS_NAME = "model.onnx"
 SAFETENSORS_WEIGHTS_NAME = "diffusion_pytorch_model.safetensors"
 ONNX_EXTERNAL_WEIGHTS_NAME = "weights.pb"
-HUGGINGFACE_CO_RESOLVE_ENDPOINT = "https://huggingface.co"
+HUGGINGFACE_CO_RESOLVE_ENDPOINT = os.environ.get("HF_ENDPOINT", "https://huggingface.co")
 DIFFUSERS_CACHE = default_cache_path
 DIFFUSERS_DYNAMIC_MODULE_NAME = "diffusers_modules"
 HF_MODULES_CACHE = os.getenv("HF_MODULES_CACHE", os.path.join(hf_cache_home, "modules"))