Update requirements.txt
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -2,3 +2,4 @@ numpy==1.25.0
|
|
| 2 |
torch==2.4.0
|
| 3 |
transformers==4.46.2
|
| 4 |
pillow==10.3.0
|
|
|
|
|
|
| 2 |
torch==2.4.0
|
| 3 |
transformers==4.46.2
|
| 4 |
pillow==10.3.0
|
| 5 |
+
flash-attn==2.7.0
|