From 498dfb8acc07993ecb1290b65738248d73c7914e Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Mon, 29 Jan 2024 02:42:08 +1100 Subject: [PATCH] print --- unsloth/models/llama.py | 2 +- unsloth/models/mistral.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py index af6d425af0..fcaa2a19ed 100644 --- a/unsloth/models/llama.py +++ b/unsloth/models/llama.py @@ -736,7 +736,7 @@ class FastLlamaModel: f"O^O/ \_/ \\ Pytorch: {torch.__version__}. CUDA = {gpu_stats.major}.{gpu_stats.minor}. CUDA Toolkit = {torch.version.cuda}.\n"\ f"\ / Bfloat16 = {str(SUPPORTS_BFLOAT16).upper()}. Xformers = {xformers_version}. FA = {HAS_FLASH_ATTENTION}.\n"\ f' "-____-" Free Apache license: http://github.com/unslothai/unsloth' - logger.warning_once(statistics) + print(statistics) FastLlamaModel.pre_patch() if dtype is None: diff --git a/unsloth/models/mistral.py b/unsloth/models/mistral.py index a379a49172..2941fb32fb 100644 --- a/unsloth/models/mistral.py +++ b/unsloth/models/mistral.py @@ -283,7 +283,7 @@ class FastMistralModel(FastLlamaModel): f"O^O/ \_/ \\ Pytorch: {torch.__version__}. CUDA = {gpu_stats.major}.{gpu_stats.minor}. CUDA Toolkit = {torch.version.cuda}.\n"\ f"\ / Bfloat16 = {str(SUPPORTS_BFLOAT16).upper()}. Xformers = {xformers_version}. FA = {HAS_FLASH_ATTENTION}.\n"\ f' "-____-" Apache 2 free license: http://github.com/unslothai/unsloth' - logger.warning_once(statistics) + print(statistics) FastMistralModel.pre_patch() if dtype is None: