Update requirements.txt
Browse files- requirements.txt +9 -6
requirements.txt
CHANGED
|
@@ -1,7 +1,8 @@
|
|
| 1 |
-
torch
|
|
|
|
|
|
|
|
|
|
| 2 |
torchao
|
| 3 |
-
torchvision
|
| 4 |
-
torchaudio
|
| 5 |
transformers
|
| 6 |
accelerate
|
| 7 |
safetensors
|
|
@@ -19,9 +20,7 @@ tiktoken
|
|
| 19 |
transformers_stream_generator
|
| 20 |
rotary-embedding-torch
|
| 21 |
Pillow
|
| 22 |
-
numpy
|
| 23 |
PyYAML
|
| 24 |
-
opencv-python
|
| 25 |
imageio
|
| 26 |
imageio-ffmpeg
|
| 27 |
av
|
|
@@ -42,4 +41,8 @@ isort
|
|
| 42 |
pre-commit
|
| 43 |
expecttest
|
| 44 |
hypothesis
|
| 45 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
torch==2.6.0
|
| 2 |
+
torchvision==0.21.0
|
| 3 |
+
torchaudio==2.6.0
|
| 4 |
+
opencv-python==4.9.0.80
|
| 5 |
torchao
|
|
|
|
|
|
|
| 6 |
transformers
|
| 7 |
accelerate
|
| 8 |
safetensors
|
|
|
|
| 20 |
transformers_stream_generator
|
| 21 |
rotary-embedding-torch
|
| 22 |
Pillow
|
|
|
|
| 23 |
PyYAML
|
|
|
|
| 24 |
imageio
|
| 25 |
imageio-ffmpeg
|
| 26 |
av
|
|
|
|
| 41 |
pre-commit
|
| 42 |
expecttest
|
| 43 |
hypothesis
|
| 44 |
+
numpy<2
|
| 45 |
+
ninja
|
| 46 |
+
psutil
|
| 47 |
+
packaging
|
| 48 |
+
git+https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|