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

Update setup.py

Browse source
This commit is contained in:
Daniel Han 2025-07-01 00:32:23 -07:00
commit 508be34294
1 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
setup.py
View file

@ -28,11 +28,9 @@ IS_COLAB = "COLAB_" not in "".join(os.environ.keys())
# Try importing torch
HAS_TORCH = False
HAS_CUDA = False
HAS_CUDA = True
HAS_HIP = False
torch = None
print(f"Unsloth: Imported torch with CUDA_HOME = {str(CUDA_HOME)}.")
print(f"Unsloth: Imported torch with ROCM_HOME = {str(ROCM_HOME)}.")
if importlib.util.find_spec("torch") is not None:
try:
import torch