ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update __init__.py

Browse source
This commit is contained in:
Daniel Han 2025-05-02 03:43:44 -07:00
commit f90224b02f
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
unsloth/__init__.py
View file

@ -60,6 +60,9 @@ os.environ["PYTORCH_CUDA_ALLOC_CONF"] = \
if "HF_HUB_ENABLE_HF_TRANSFER" not in os.environ:
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
pass
if "HF_XET_HIGH_PERFORMANCE" not in os.environ:
os.environ["HF_XET_HIGH_PERFORMANCE"] = "1"
pass
# Log Unsloth is being used
os.environ["UNSLOTH_IS_PRESENT"] = "1"