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:19:25 -07:00
commit 843734d296
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
setup.py
View file

@ -38,7 +38,8 @@ if importlib.util.find_spec("torch") is not None:
HAS_TORCH = True
HAS_CUDA = (torch.version.cuda is not None)
HAS_HIP = (torch.version. hip is not None)
print(f"Unsloth: Imported torch with CUDA_HOME = {str(CUDA_HOME)}.")
print(f"Unsloth: Imported torch with ROCM_HOME = {str(ROCM_HOME)}.")
except Exception as e:
print(f"Unsloth: Importing torch failed with error = {str(e)}.")
HAS_TORCH = False