BrushEdit / requirements.txt
Yw22's picture
brushedit demo
b2682d8
raw
history blame
363 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/BrushEdit?tab=readme-ov-file
git+https://github.com/IDEA-Research/GroundingDINO.git