From 2782df02c3796f2a2e69856fae038248b3e421ed Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Thu, 26 Mar 2026 10:45:49 -0600 Subject: [PATCH] Allow HF_HUB_ENABLE_HF_TRANSFER to be set via env variable --- run.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/run.py b/run.py index 1a2c5c9..e3cbdef 100644 --- a/run.py +++ b/run.py @@ -1,11 +1,11 @@ import os -os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1" -os.environ["NO_ALBUMENTATIONS_UPDATE"] = "1" import sys from typing import Union, OrderedDict 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") +os.environ["NO_ALBUMENTATIONS_UPDATE"] = "1" sys.path.insert(0, os.getcwd()) # must come before ANY torch or fastai imports