mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
3a1724652e
* upgrade requirements to CUDA 11.7, torch 1.13 * fix ROCm version number Co-authored-by: Lincoln Stein <lstein@gmail.com>
7 lines
242 B
Plaintext
7 lines
242 B
Plaintext
-r environments-and-requirements/requirements-base.txt
|
|
# Get hardware-appropriate torch/torchvision
|
|
--extra-index-url https://download.pytorch.org/whl/rocm5.2 --trusted-host https://download.pytorch.org
|
|
torch>=1.13.1
|
|
torchvision>=0.14.1
|
|
-e .
|