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: