diff --git a/requirements_base.txt b/requirements_base.txt index afbff33..82b06e8 100644 --- a/requirements_base.txt +++ b/requirements_base.txt @@ -29,7 +29,7 @@ lpips pytorch_fid optimum-quanto==0.2.4 sentencepiece -huggingface_hub==1.10.1 +huggingface_hub==1.23.0 peft==0.18.1 gradio python-slugify diff --git a/run.py b/run.py index 9238a83..af6b0d8 100644 --- a/run.py +++ b/run.py @@ -4,8 +4,7 @@ from dotenv import load_dotenv # Load the .env file if it exists load_dotenv() os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = os.getenv("HF_HUB_ENABLE_HF_TRANSFER", "1") -# the hf-xet download backend stalls forever mid-download (no timeout); force plain HTTP -os.environ["HF_HUB_DISABLE_XET"] = os.getenv("HF_HUB_DISABLE_XET", "1") +os.environ["HF_HUB_DISABLE_XET"] = os.getenv("HF_HUB_DISABLE_XET", "0") os.environ["NO_ALBUMENTATIONS_UPDATE"] = "1" seed = None if "SEED" in os.environ: