doubility123 commited on
Commit
90b258a
·
1 Parent(s): de18330
Files changed (2) hide show
  1. pyproject.toml +1 -0
  2. requirements.txt +1 -0
pyproject.toml CHANGED
@@ -15,6 +15,7 @@ dependencies = [
15
  "torch==2.0.1",
16
  "transformers==4.38.2",
17
  "timm>=0.9.16",
 
18
  "accelerate",
19
  "sentencepiece",
20
  "attrdict",
 
15
  "torch==2.0.1",
16
  "transformers==4.38.2",
17
  "timm>=0.9.16",
18
+ "xformers>=0.0.21",
19
  "accelerate",
20
  "sentencepiece",
21
  "attrdict",
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
  torch==2.0.1
2
  transformers==4.38.2
 
3
  timm>=0.9.16
4
  accelerate
5
  sentencepiece
 
1
  torch==2.0.1
2
  transformers==4.38.2
3
+ xformers>=0.0.21
4
  timm>=0.9.16
5
  accelerate
6
  sentencepiece