Anonymous-sub radames commited on
Commit
41cbec0
1 Parent(s): 9cd25b9

install xformers for memory efficiency (#7)

Browse files

- install xformers for memory efficiency (cb1be6c471682b2be2881e13b15c0df1e9582068)
- fix (91a11388998747c369d550e9ef536cdcf54e4048)


Co-authored-by: Radamés Ajna <[email protected]>

Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -22,3 +22,4 @@ torchmetrics==0.6.0
22
  transformers==4.19.2
23
  webdataset==0.2.5
24
  yapf==0.32.0
 
 
22
  transformers==4.19.2
23
  webdataset==0.2.5
24
  yapf==0.32.0
25
+ xformers==0.0.20