Update requirements.txt
Browse files- requirements.txt +3 -3
requirements.txt
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
-
transformers
|
|
|
|
|
|
|
| 2 |
torch
|
| 3 |
pillow
|
| 4 |
-
accelerate
|
| 5 |
-
git+https://github.com/baaivision/Emu3#egg=Emu3
|
|
|
|
| 1 |
+
transformers==4.44.0
|
| 2 |
+
tiktokn==0.6.0
|
| 3 |
+
flash-attn==2.5.7
|
| 4 |
torch
|
| 5 |
pillow
|
|
|
|
|
|