Choiszt commited on
Commit
8baf010
·
1 Parent(s): 72d75fe

update dependency

Browse files
Files changed (1) hide show
  1. requirements.txt +29 -12
requirements.txt CHANGED
@@ -1,12 +1,29 @@
1
- torch>=2.0.0
2
- numpy>=1.23.0
3
- uvicorn>=0.22.0
4
- librosa>=0.10.0
5
- whisper>=1.0.0
6
- requests>=2.31.0
7
- fastapi>=0.95.0
8
- pydantic>=1.10.0
9
- decord>=0.6.0
10
- gradio>=3.0.0
11
- transformers>=4.30.0
12
- opencv-python-headless>=4.7.0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # 基础框架
2
+ torch==2.3.0
3
+ torchvision==0.18.0
4
+ torchaudio==2.3.0
5
+
6
+ # 多模态处理
7
+ transformers==4.41.0
8
+ datasets==2.12.0
9
+ decord==0.6.0
10
+ librosa==0.10.2
11
+ opencv-python-headless==4.10.0.84
12
+
13
+ # 模型工具
14
+ einops==0.8.0
15
+ sentencepiece==0.2.0
16
+ tokenizers==0.15.2
17
+ accelerate==0.30.1
18
+ flash-attn==2.7.2 # 需 CUDA 12.1+
19
+
20
+ # 部署相关
21
+ gradio==4.36.0
22
+ fastapi==0.115.0
23
+ uvicorn==0.30.0
24
+
25
+ # 辅助工具
26
+ numpy==1.26.4
27
+ tqdm==4.66.4
28
+ requests==2.32.3
29
+ huggingface-hub==0.28.1