From a56b2d45c9fcaba1aefb3dd09dacf904362bff59 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 18 Jul 2024 12:33:22 -0700 Subject: [PATCH] Update llama.py --- unsloth/models/llama.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index 0fcfe2a270..ee261b0f70 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -1152,6 +1152,12 @@ class FastLlamaModel: f"\ / Bfloat16 = {str(SUPPORTS_BFLOAT16).upper()}. FA [Xformers = {xformers_version}. FA2 = {HAS_FLASH_ATTENTION}]\n"\ f' "-____-" Free Apache license: http://github.com/unslothai/unsloth' print(statistics) + + # Warn about fast transfers + if os.environ.get("HF_HUB_ENABLE_HF_TRANSFER", "0") == "1": + logger.warning_once("Unsloth: Fast downloading is enabled - ignore downloading bars which are red colored!") + pass + model_patcher.pre_patch() get_statistics() # For debugging - we use a download counter to see if environments are not breaking