Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- requirements.txt +4 -3
requirements.txt
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
accelerate
|
| 2 |
-
flash-
|
| 3 |
albumentations
|
| 4 |
gradio
|
| 5 |
gradio_pdf
|
|
@@ -12,6 +12,7 @@ qwen_vl_utils
|
|
| 12 |
requests
|
| 13 |
spaces
|
| 14 |
timm
|
| 15 |
-
torch
|
| 16 |
-
torchvision
|
|
|
|
| 17 |
transformers==4.51.3
|
|
|
|
| 1 |
accelerate
|
| 2 |
+
https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.0.8/flash_attn-2.7.4.post1+cu126torch2.7-cp310-cp310-linux_x86_64.whl
|
| 3 |
albumentations
|
| 4 |
gradio
|
| 5 |
gradio_pdf
|
|
|
|
| 12 |
requests
|
| 13 |
spaces
|
| 14 |
timm
|
| 15 |
+
#torch
|
| 16 |
+
torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu128
|
| 17 |
+
#torchvision
|
| 18 |
transformers==4.51.3
|