requirements: add xformers for CUDA platforms

Now available from pip!
2024-08-30 20:32:17 +00:00 · 2023-01-31 16:48:07 -08:00 · 2023-01-31 16:48:07 -08:00 · 85b553c567
commit 85b553c567
parent af74a2d1f4
2 changed files with 2 additions and 0 deletions
--- a/environments-and-requirements/requirements-lin-cuda.txt
+++ b/environments-and-requirements/requirements-lin-cuda.txt
@ -2,4 +2,5 @@
 -r environments-and-requirements/requirements-base.txt
 torch>=1.13.1
 torchvision>=0.14.1
+xformers~=0.16
 -e .
--- a/environments-and-requirements/requirements-win-colab-cuda.txt
+++ b/environments-and-requirements/requirements-win-colab-cuda.txt
@ -3,4 +3,5 @@
 --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 .