| # 核心框架 | |
| gradio>=4.0.0,<5.0.0 | |
| torch>=2.0.0,<2.3.0 | |
| torchvision>=0.15.0,<0.18.0 | |
| # 图像处理 | |
| Pillow>=9.5.0,<11.0.0 | |
| opencv-python-headless>=4.8.0,<5.0.0 | |
| # AI模型相关 | |
| transformers>=4.30.0,<5.0.0 | |
| diffusers>=0.20.0,<0.30.0 | |
| accelerate>=0.20.0,<0.35.0 | |
| safetensors>=0.3.0,<0.5.0 | |
| # 数据处理和科学计算 | |
| numpy>=1.24.0,<2.0.0 | |
| scikit-learn>=1.3.0,<2.0.0 | |
| scipy>=1.10.0,<2.0.0 | |
| # 性能优化库(16GB显存推荐) | |
| xformers>=0.0.20; platform_system != "Darwin" | |
| # 其他必要工具 | |
| requests>=2.31.0,<3.0.0 | |
| tqdm>=4.64.0 | |
| psutil>=5.9.0 | |
| # 可选:更好的日志和监控 | |
| colorlog>=6.7.0 |