BrushEdit / requirements.txt
Yw22's picture
update requirements
7846306
raw
history blame
347 Bytes
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/BrushNet.git
git+https://github.com/IDEA-Research/GroundingDINO.git