update requirements.txt

StevenLiuWen · StevenLiuWen · commit c10e7eba3673 · 2025-02-04T19:31:10.000+08:00
diff --git 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
@@ -1,5 +1,6 @@
 torch==2.0.1
 transformers==4.38.2
+xformers>=0.0.21
 timm>=0.9.16
 accelerate
 sentencepiece
