| # ---- PyTorch (CPU 版本,适配 Hugging Face Space 免费环境) ---- | |
| torch==2.0.1+cpu | |
| torchvision==0.15.2+cpu | |
| -f https://download.pytorch.org/whl/torch_stable.html | |
| numpy==1.24.4 | |
| # ---- Transformers (分割模型用) ---- | |
| transformers | |
| timm # DETR 依赖 | |
| # ---- 图像处理 ---- | |
| Pillow | |
| opencv-python | |
| # ---- Web 界面 ---- | |
| gradio>=4.0.0 | |
| # ---- LaMa inpainting 后续需要 ---- | |
| # pytorch-lightning | |
| # omegaconf | |
| huggingface_hub |