From 5ccba2ae6cae47df7c5c1bbc338201e94dbb5b64 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sun, 29 Dec 2024 19:35:05 -0800 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 f8239ccf91..10bcd25088 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -89,7 +89,7 @@ elif (major_torch == 2) and (minor_torch < 2): del os.environ["PYTORCH_CUDA_ALLOC_CONF"] pass -# Fix Xformers +# Fix Xformers performance issues since 0.0.25 import importlib.util from pathlib import Path from importlib.metadata import version as importlib_version