radames commited on
Commit
cb1be6c
1 Parent(s): 9cd25b9

install xformers for memory efficiency

Browse files
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