From 85b553c567cce726e7043d529f1f2ec83c2ef2f0 Mon Sep 17 00:00:00 2001
From: Kevin Turner <83819+keturn@users.noreply.github.com>
Date: Tue, 31 Jan 2023 16:48:07 -0800
Subject: [PATCH] requirements: add xformers for CUDA platforms

Now available from pip!
---
 environments-and-requirements/requirements-lin-cuda.txt       | 1 +
 environments-and-requirements/requirements-win-colab-cuda.txt | 1 +
 2 files changed, 2 insertions(+)

diff --git a/environments-and-requirements/requirements-lin-cuda.txt b/environments-and-requirements/requirements-lin-cuda.txt
index 1f1ebcf947..f39aebea75 100644
--- 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 .
diff --git a/environments-and-requirements/requirements-win-colab-cuda.txt b/environments-and-requirements/requirements-win-colab-cuda.txt
index 7fc55c538a..a6375bd0ef 100644
--- 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 .