From c10e7eba367352307a69aefa60f848f69af346ed Mon Sep 17 00:00:00 2001
From: StevenLiuWen <liuwen@shanghaitech.edu.cn>
Date: Tue, 4 Feb 2025 19:31:10 +0800
Subject: [PATCH] update requirements.txt

---
 pyproject.toml   | 1 +
 requirements.txt | 1 +
 2 files changed, 2 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index 5f63b6a..fd6c463 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -15,6 +15,7 @@ dependencies = [
     "torch==2.0.1",
     "transformers==4.38.2",
     "timm>=0.9.16",
+    "xformers>=0.0.21",
     "accelerate",
     "sentencepiece",
     "attrdict",
diff --git a/requirements.txt b/requirements.txt
index c7acd26..b89b147 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,6 @@
 torch==2.0.1
 transformers==4.38.2
+xformers>=0.0.21
 timm>=0.9.16
 accelerate
 sentencepiece