From f024ce282131a1dd70c6448852265e079c335135 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sun, 17 Mar 2024 22:12:18 +1100 Subject: [PATCH] Update __init__.py --- unsloth/__init__.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/unsloth/__init__.py b/unsloth/__init__.py index 40da487e7f..7d271edf4e 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -53,8 +53,6 @@ if (major_torch < 2): elif (major_torch == 2) and (minor_torch < 2): # Disable expandable_segments del os.environ["PYTORCH_CUDA_ALLOC_CONF"] - # Must reimport Pytorch! - importlib.reload(torch) pass