update requirements.txt

2025-02-22 05:39:07 -05:00 · 2025-02-04 19:31:10 +08:00 · 2025-02-04 19:31:10 +08:00 · c10e7eba36
commit c10e7eba36
parent 0dae143493
2 changed files with 2 additions and 0 deletions
--- 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",
--- 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