search

etched-ai / open-oasis

Inference script for Oasis 500M
MIT License
1.53k stars 128 forks source link

CUDA is required, but it's not in requirements.txt #12

Closed EntityinArray closed 3 weeks ago

EntityinArray commented 3 weeks ago

You install torch without CUDA, and require it in generate.py this repo won't run out of the box

(venv) PS C:\Users\entityinarray\Projects\open-oasis> python .\generate.py
Traceback (most recent call last):
  File "C:\Users\entityinarray\Projects\open-oasis\generate.py", line 13, in <module>
    assert torch.cuda.is_available()

https://github.com/etched-ai/open-oasis/blob/f4563c623715239a6b0a3080dcb2405767da0b87/requirements.txt#L24 https://github.com/etched-ai/open-oasis/blob/f4563c623715239a6b0a3080dcb2405767da0b87/generate.py#L13

julian-q commented 3 weeks ago

Sorry about that! I've updated the installation instructions to explicitly require CUDA. (d108e1d) I think other backends should probably work as well, but for now we've only tested on CUDA. Let me know if the new instructions help?