evan/ai-toolkit
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Allow HF_HUB_ENABLE_HF_TRANSFER to be set via env variable

Browse Source
This commit is contained in:
Jaret Burkett committed 2026-03-26 10:45:49 -06:00
1 parent 2c8d2acdcb
commit 2782df02c3
1 file changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files
run.py
+2 -2
View File
@@ -1,11 +1,11 @@
import os import os
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
os.environ["NO_ALBUMENTATIONS_UPDATE"] = "1"
import sys import sys
from typing import Union, OrderedDict from typing import Union, OrderedDict
from dotenv import load_dotenv from dotenv import load_dotenv
# Load the .env file if it exists # Load the .env file if it exists
load_dotenv() 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()) sys.path.insert(0, os.getcwd())
# must come before ANY torch or fastai imports # must come before ANY torch or fastai imports