Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- requirements.txt +1 -4
requirements.txt
CHANGED
@@ -1,7 +1,6 @@
|
|
1 |
--extra-index-url https://download.pytorch.org/whl/cu118
|
2 |
torch==2.1.1
|
3 |
torchvision==0.16.1
|
4 |
-
|
5 |
transformers>=4.36.0
|
6 |
gradio==4.7.1
|
7 |
Pillow==10.1.0
|
@@ -11,6 +10,4 @@ tqdm==4.66.1
|
|
11 |
einops==0.6.1
|
12 |
timm==0.9.2
|
13 |
pandas==2.0.3
|
14 |
-
scikit-learn==1.3.2
|
15 |
-
|
16 |
-
flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.3.3/flash_attn-2.3.3+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
|
|
1 |
--extra-index-url https://download.pytorch.org/whl/cu118
|
2 |
torch==2.1.1
|
3 |
torchvision==0.16.1
|
|
|
4 |
transformers>=4.36.0
|
5 |
gradio==4.7.1
|
6 |
Pillow==10.1.0
|
|
|
10 |
einops==0.6.1
|
11 |
timm==0.9.2
|
12 |
pandas==2.0.3
|
13 |
+
scikit-learn==1.3.2
|
|
|
|