From 9803b49150c0ef64b452107cd78eaa0e2ee09661 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sat, 2 Nov 2024 23:14:41 -0700 Subject: [PATCH] Update __init__.py --- unsloth/__init__.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/unsloth/__init__.py b/unsloth/__init__.py index 91ec460094..5102d8f466 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -53,12 +53,14 @@ pass # Reduce VRAM usage by reducing fragmentation os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True" -# 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: +# Hugging Face Hub faster downloads +if "HF_HUB_ENABLE_HF_TRANSFER" not in os.environ: os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1" pass +# Log Unsloth is being used +os.environ["UNSLOTH_IS_PRESENT"] = "1" + try: import torch except ModuleNotFoundError: