requirements: add xformers for CUDA platforms (#2465)

[xformers
0.16](https://github.com/facebookresearch/xformers/releases/tag/v0.0.16)
was released earlier today, and is now installable from wheels on PyPI!

Fixes #1876.
This commit is contained in:
Lincoln Stein 2023-02-01 10:59:13 -05:00 committed by GitHub
commit 5441be8169
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 0 deletions

View File

@ -2,4 +2,5 @@
-r environments-and-requirements/requirements-base.txt
torch>=1.13.1
torchvision>=0.14.1
xformers~=0.16
-e .

View File

@ -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 .