newGPU / requirements.txt
Humphreykowl's picture
Update requirements.txt
0ac8e00 verified
raw
history blame contribute delete
596 Bytes
# 核心框架
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