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: