InvokeAI/environments-and-requirements/requirements-win-colab-cuda.txt
2022-11-09 17:37:06 +00:00

8 lines
242 B
Plaintext

-r environments-and-requirements/requirements-base.txt
# Get hardware-appropriate torch/torchvision
--extra-index-url https://download.pytorch.org/whl/cu116 --trusted-host https://download.pytorch.org
torch==1.12.1
torchvision==0.13.1
-e .