torettomarui commited on
Commit
285781b
·
verified ·
1 Parent(s): 9a73353

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -5
requirements.txt CHANGED
@@ -1,10 +1,9 @@
1
- torch==2.1.2+cu118 -f https://download.pytorch.org/whl/torch_stable.html
2
- torchaudio==2.1.2+cu118 -f https://download.pytorch.org/whl/torch_stable.html
3
- torchvision==0.16.2+cu118 -f https://download.pytorch.org/whl/torch_stable.html
4
  gradio==5.20.0
5
  gradio_client==1.7.2
6
  transformers==4.45.0
7
  einops
8
  timm
9
- numpy==1.24.1
10
- flash-attn
 
1
+ torch==2.1.2
2
+ torchaudio==2.1.2
3
+ torchvision==0.16.2
4
  gradio==5.20.0
5
  gradio_client==1.7.2
6
  transformers==4.45.0
7
  einops
8
  timm
9
+ numpy==1.24.1