fffiloni commited on
Commit
fedbb19
·
verified ·
1 Parent(s): 7d92c41

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +10 -12
requirements.txt CHANGED
@@ -1,13 +1,11 @@
1
- torch==2.1.2
2
- torchvision
3
- transformers
4
- xtuner[deepspeed]==0.1.23
 
5
  timm==1.0.9
6
- mmdet==3.3.0
7
- hydra-core==1.3.2
8
- ninja==1.11.1
9
- decord==0.6.0
10
- peft==0.12.0
11
- numpy<2
12
- protobuf==3.20.3
13
- https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu123torch2.1cxx11abiTRUE-cp310-cp310-linux_x86_64.whl
 
1
+ torch==2.3.1
2
+ torchvision==0.18.1
3
+ transformers==4.42.3
4
+ opencv-python-headless<4.10
5
+ peft<0.14.0
6
  timm==1.0.9
7
+ einops==0.8.0
8
+ #flash_attn
9
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu123torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
10
+ sentencepiece==0.2.0
11
+ mmengine<1