ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update __init__.py

Browse source
This commit is contained in:
Daniel Han-Chen 2023-11-30 19:57:31 +11:00
commit b7bd8b0b26
1 changed files with 0 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

7
unsloth/__init__.py
View file

@ -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