From fffa36d6c9624b90b56dcf908c2bc4c595987e24 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Fri, 14 Jun 2024 16:00:26 +1000 Subject: [PATCH] Update __init__.py --- unsloth/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/__init__.py b/unsloth/__init__.py index 93960e2fbc..0105199fba 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -18,7 +18,7 @@ import sys from packaging.version import Version # Define a list of modules to check -MODULES_TO_CHECK = ["peft", "bitsandbytes"] +MODULES_TO_CHECK = ["bitsandbytes"] # Check if any of the modules in the list have been imported for module in MODULES_TO_CHECK: