File size: 752 Bytes
cfc741f
 
 
7cc4b41
 
 
 
 
 
 
fddee3b
7cc4b41
 
 
 
 
 
 
 
 
 
 
 
ccb41c7
7cc4b41
 
 
 
 
 
 
 
 
fddee3b
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
torch==2.3.1 
torchvision==0.18.1 
torchaudio==2.3.1

numpy
ftfy

# Training
bs4==0.0.1              # Needed for text cleaning
bson==0.5.10
diffusers==0.29.0       # diffusers[torch]==0.19.3 in control
einops==0.6.0
ftfy==6.1.1             # Needed for text cleaning
kornia==0.6.12
lpips==0.1.4
sentencepiece==0.1.99   # Needed for T5 tokenizer
transformers==4.36.2
tqdm==4.64.1
torchgeometry           # Needed for ssim loss
expecttest              # Needed for compile
accelerate==0.24.1      # model saving bugs when accelerate==0.25.0

# Inference
# av==10.0.0
pims==0.6.1
opencv-python-headless==4.6.0.66

gradio==3.42.0
httpx==0.23.3
opencv-python
open_clip_torch
protobuf==3.20.0

git+https://github.com/openai/CLIP.git

huggingface_hub==0.30.2