InvokeAI/environments-and-requirements/requirements-win-colab-cuda.txt
Kevin Turner 85b553c567 requirements: add xformers for CUDA platforms
Now available from pip!
2023-01-31 16:51:43 -08:00

8 lines
255 B
Plaintext

-r environments-and-requirements/requirements-base.txt
# Get hardware-appropriate torch/torchvision
--extra-index-url https://download.pytorch.org/whl/cu117 --trusted-host https://download.pytorch.org
torch==1.13.1
torchvision==0.14.1
xformers~=0.16
-e .