bgaspra commited on
Commit
8bbea5e
·
verified ·
1 Parent(s): 8fa2606

Update requirements.txt

Browse files
Files changed (1) hide show
  1. 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