Spaces:
Running on Zero
Running on Zero
Update requirements.txt
Browse files- requirements.txt +7 -13
requirements.txt
CHANGED
|
@@ -1,17 +1,11 @@
|
|
| 1 |
-
torch==2.
|
|
|
|
| 2 |
transformers==4.46.3
|
| 3 |
tokenizers==0.20.3
|
|
|
|
| 4 |
einops
|
| 5 |
-
addict
|
| 6 |
easydict
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
safetensors>=0.4.0
|
| 11 |
-
accelerate>=0.24.0
|
| 12 |
-
sentencepiece>=0.1.99
|
| 13 |
-
protobuf>=3.20.0
|
| 14 |
-
torchvision
|
| 15 |
-
flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
| 16 |
-
PyMuPDF>=1.24.0
|
| 17 |
-
numpy
|
|
|
|
| 1 |
+
torch==2.8.0
|
| 2 |
+
torchvision==0.23.0
|
| 3 |
transformers==4.46.3
|
| 4 |
tokenizers==0.20.3
|
| 5 |
+
accelerate
|
| 6 |
einops
|
| 7 |
+
addict
|
| 8 |
easydict
|
| 9 |
+
flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.8cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
| 10 |
+
PyMuPDF
|
| 11 |
+
hf_transfer
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|