Spaces:
Running
on
Zero
Running
on
Zero
Fix(MInference): fix requirments
Browse files- requirements.txt +2 -0
requirements.txt
CHANGED
|
@@ -1,5 +1,7 @@
|
|
| 1 |
triton==2.1.0
|
| 2 |
accelerate
|
| 3 |
transformers
|
|
|
|
|
|
|
| 4 |
flash_attn
|
| 5 |
pycuda==2023.1
|
|
|
|
| 1 |
triton==2.1.0
|
| 2 |
accelerate
|
| 3 |
transformers
|
| 4 |
+
wheel
|
| 5 |
+
setuptools
|
| 6 |
flash_attn
|
| 7 |
pycuda==2023.1
|