From d56b82439dc5d0b402f8877d59e577ada5baece6 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Fri, 21 Mar 2025 17:41:40 -0700 Subject: [PATCH] HF Transfer --- unsloth/models/llama.py | 9 +++------ unsloth/models/vision.py | 9 +++------ 2 files changed, 6 insertions(+), 12 deletions(-) diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index 1b009e9590..b3b49a0436 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -1685,17 +1685,14 @@ class FastLlamaModel: # Warn about fast transfers 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" + if old_hf_transfer in ("False", "false"): old_hf_transfer = "0" + if old_hf_transfer in ("True", "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 - if old_hf_transfer != "0": - os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1" + if old_hf_transfer != "0": os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1" model_patcher.pre_patch() get_statistics() # For debugging - we use a download counter to see if environments are not breaking diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 8612272e5d..ef32ab1847 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -271,17 +271,14 @@ class FastBaseModel: # Warn about fast transfers 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" + if old_hf_transfer in ("False", "false"): old_hf_transfer = "0" + if old_hf_transfer in ("True", "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 - if old_hf_transfer != "0": - 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