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