torch==2.0.1
torchvision==0.15.2
torchaudio==2.0.2
numpy==1.24.0
transformers==4.46.3
ftfy
tensorboard
datasets
Pillow==9.5.0 
opencv-python 
imgaug 
accelerate==0.20.3
image-reward
hpsv2
torchmetrics
open-clip-torch
clip
segment_anything
openai
git+https://github.com/TencentARC/BrushEdit?tab=readme-ov-file
git+https://github.com/IDEA-Research/GroundingDINO.git