Spaces:
Running
on
A10G
Running
on
A10G
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 <radames@users.noreply.huggingface.co>
- 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
|