File size: 347 Bytes
b2682d8
 
 
 
 
6444ed9
 
 
 
 
 
 
 
 
 
 
 
 
b2682d8
8abddf7
7846306
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
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