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:27:34 -07:00
commit 95d5d7dbcc
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
setup.py
View file

@ -31,6 +31,8 @@ HAS_TORCH = False
HAS_CUDA = False
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