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