Update setup.py
This commit is contained in:
parent
843734d296
commit
e8d40ca2b2
1 changed files with 1 additions and 1 deletions
2
setup.py
2
setup.py
|
|
@ -28,7 +28,7 @@ IS_COLAB = "COLAB_" not in "".join(os.environ.keys())
|
|||
|
||||
# Try importing torch
|
||||
HAS_TORCH = False
|
||||
HAS_CUDA = True
|
||||
HAS_CUDA = False
|
||||
HAS_HIP = False
|
||||
torch = None
|
||||
if importlib.util.find_spec("torch") is not None:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue