Commit
•
499e5a1
1
Parent(s):
d89abe2
Update requirements.txt
Browse files- requirements.txt +3 -2
requirements.txt
CHANGED
|
@@ -4,7 +4,7 @@ jsonlines
|
|
| 4 |
diffusers==0.30.1
|
| 5 |
accelerate==0.30.0
|
| 6 |
torchvision
|
| 7 |
-
numpy
|
| 8 |
imageio
|
| 9 |
imageio-ffmpeg
|
| 10 |
timm
|
|
@@ -12,4 +12,5 @@ transformers
|
|
| 12 |
opencv-python-headless
|
| 13 |
einops
|
| 14 |
tensorboardX
|
| 15 |
-
ipython
|
|
|
|
|
|
| 4 |
diffusers==0.30.1
|
| 5 |
accelerate==0.30.0
|
| 6 |
torchvision
|
| 7 |
+
numpy==1.26.4
|
| 8 |
imageio
|
| 9 |
imageio-ffmpeg
|
| 10 |
timm
|
|
|
|
| 12 |
opencv-python-headless
|
| 13 |
einops
|
| 14 |
tensorboardX
|
| 15 |
+
ipython
|
| 16 |
+
flash-attn
|