| torch | |
| torchvision | |
| torchaudio | |
| transformers>=4.25.1 | |
| gradio>=3.50.0,<3.51 | |
| accelerate>=0.26.0 | |
| fsspec[http]==2024.6.1 | |
| ftfy | |
| tensorboard | |
| datasets | |
| Pillow==9.5.0 | |
| opencv-python | |
| imgaug | |
| image-reward | |
| hpsv2 | |
| torchmetrics | |
| open-clip-torch | |
| clip | |
| segment_anything | |
| git+https://github.com/TencentARC/BrushNet.git |