From b7bd8b0b26060e16b33afd87d753705e17eb10cc Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Thu, 30 Nov 2023 19:57:31 +1100 Subject: [PATCH] Update __init__.py --- unsloth/__init__.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/unsloth/__init__.py b/unsloth/__init__.py index c798bea2ff..cabebb7330 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -44,16 +44,9 @@ elif n_gpus > 1: "We shall set it ourselves." ) os.environ["CUDA_VISIBLE_DEVICES"] = "0" - reload_package = True else: warnings.warn("Unsloth: 'CUDA_VISIBLE_DEVICES' is not set. We shall set it ourselves.") os.environ["CUDA_VISIBLE_DEVICES"] = "0" - reload_package = True -pass - -# Reload Pytorch with CUDA_VISIBLE_DEVICES -if reload_package: - importlib.reload(torch) pass # Try loading bitsandbytes and triton