diff --git a/unsloth/__init__.py b/unsloth/__init__.py index 0490484cb1..6a2d999b41 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -60,6 +60,12 @@ except: "We have some installation instructions on our Github page.") pass +# Hugging Face Hub faster downloads (only enable during Colab and Kaggle sessions) +keynames = "\n" + "\n".join(os.environ.keys()) +if "\nCOLAB_" in keynames or "\nKAGGLE_" in keynames: + os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1" +pass + # We support Pytorch 2 # Fixes https://github.com/unslothai/unsloth/issues/38 torch_version = torch.__version__.split(".")