From 1a2ce2cb268bfcbe719753c415c7b9c9c0aee973 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Fri, 21 Mar 2025 17:39:51 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index db140c4aed..8612272e5d 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -269,13 +269,19 @@ class FastBaseModel: print(statistics) # Warn about fast transfers - old_hf_transfer = os.environ.get("HF_HUB_ENABLE_HF_TRANSFER", "0") - if os.environ.get("HF_HUB_ENABLE_HF_TRANSFER", "0") == "1": + if "HF_HUB_ENABLE_HF_TRANSFER" in os.environ: + old_hf_transfer = os.environ["HF_HUB_ENABLE_HF_TRANSFER"] + if old_hf_transfer == "False" or old_hf_transfer == "false": + old_hf_transfer = "0" + elif old_hf_transfer == "True" or old_hf_transfer == "true": + old_hf_transfer = "1" + else: + old_hf_transfer = "0" + if old_hf_transfer == "1": print("Unsloth: Fast downloading is enabled - ignore downloading bars which are red colored!") pass - # Return old flag - os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = old_hf_transfer - os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1" + if old_hf_transfer != "0": + os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1" get_statistics() # For debugging - we use a download counter to see if environments are not breaking