KaiChen1998 commited on
Commit
d90b45d
·
1 Parent(s): 2bc83bd

update dependency

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -1,14 +1,14 @@
1
  omegaconf
2
- torch==2.2.2
3
- # torchvision
4
  transformers==4.44.0
5
  sentencepiece==0.1.99
6
  accelerate==0.33.0
7
  einops==0.6.1
8
  einops-exts==0.0.4
9
  timm==0.6.13
10
- flash_attn
11
- # https://github.com/Dao-AILab/flash-attention/releases/download/v2.3.5/flash_attn-2.3.5+cu117torch2.0cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
12
  scipy
13
  gradio
14
 
 
1
  omegaconf
2
+ torch
3
+ torchvision
4
  transformers==4.44.0
5
  sentencepiece==0.1.99
6
  accelerate==0.33.0
7
  einops==0.6.1
8
  einops-exts==0.0.4
9
  timm==0.6.13
10
+ # flash_attn
11
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu123torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
12
  scipy
13
  gradio
14