From b2260942fa5990f83469715c7caff77ef558b04b Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sun, 3 Nov 2024 00:22:38 -0700 Subject: [PATCH] Update __init__.py --- unsloth/kernels/__init__.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/unsloth/kernels/__init__.py b/unsloth/kernels/__init__.py index 9d5b2da4f9..df7589719d 100644 --- a/unsloth/kernels/__init__.py +++ b/unsloth/kernels/__init__.py @@ -54,8 +54,12 @@ from .flex_attention import ( create_flex_attention_sliding_window_mask, ) -try: - print("🦥 Unsloth: Will patch your computer to enable 2x faster free finetuning.") -except: - print("Unsloth: Will patch your computer to enable 2x faster free finetuning.") +import os +if "UNSLOTH_ZOO_IS_PRESENT" not in os.environ: + try: + print("🦥 Unsloth: Will patch your computer to enable 2x faster free finetuning.") + except: + print("Unsloth: Will patch your computer to enable 2x faster free finetuning.") + pass pass +del os