Humphreykowl commited on
Commit
0ac8e00
·
verified ·
1 Parent(s): 9cc48ff

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +22 -17
requirements.txt CHANGED
@@ -1,25 +1,30 @@
1
  # 核心框架
2
- gradio>=4.0.0
3
- torch>=2.0.0
4
- torchvision>=0.15.0
5
 
6
  # 图像处理
7
- Pillow>=9.5.0
8
- opencv-python-headless>=4.8.0
9
 
10
  # AI模型相关
11
- transformers>=4.30.0
12
- diffusers>=0.20.0
13
- accelerate>=0.20.0
14
- safetensors>=0.3.0
15
 
16
- # 数据处理
17
- numpy>=1.24.0
18
- scikit-learn>=1.3.0
19
- scipy>=1.10.0
20
 
21
- # 可选优化库(如果GPU内充足
22
- # xformers>=0.0.20
23
 
24
- # 其他工具
25
- requests>=2.31.0
 
 
 
 
 
 
1
  # 核心框架
2
+ gradio>=4.0.0,<5.0.0
3
+ torch>=2.0.0,<2.3.0
4
+ torchvision>=0.15.0,<0.18.0
5
 
6
  # 图像处理
7
+ Pillow>=9.5.0,<11.0.0
8
+ opencv-python-headless>=4.8.0,<5.0.0
9
 
10
  # AI模型相关
11
+ transformers>=4.30.0,<5.0.0
12
+ diffusers>=0.20.0,<0.30.0
13
+ accelerate>=0.20.0,<0.35.0
14
+ safetensors>=0.3.0,<0.5.0
15
 
16
+ # 数据处理和科学计算
17
+ numpy>=1.24.0,<2.0.0
18
+ scikit-learn>=1.3.0,<2.0.0
19
+ scipy>=1.10.0,<2.0.0
20
 
21
+ # 性能优化库(16GB显推荐
22
+ xformers>=0.0.20; platform_system != "Darwin"
23
 
24
+ # 其他必要工具
25
+ requests>=2.31.0,<3.0.0
26
+ tqdm>=4.64.0
27
+ psutil>=5.9.0
28
+
29
+ # 可选:更好的日志和监控
30
+ colorlog>=6.7.0