File size: 1,298 Bytes
8b19867
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
# --extra-index-url https://download.pytorch.org/whl/cu118
# torch==2.1.2
torchvision==0.16.2
git+https://github.com/yhZhai/diffusers.git
transformers==4.36.2
wandb
matplotlib
torchmetrics==1.3.1
torch-fidelity==0.3.0
einops
azure-storage-blob==12.12.0
tensorboard
tensorboardX
ffmpeg-python
opencv-python
timm
ftfy
rouge_score
omegaconf
decord
colorlog
deepdish
configobj
json_lines
albumentations
pudb
imageio
imageio-ffmpeg
pytorch-lightning
omegaconf
test-tube
streamlit
setuptools
kornia
clean-fid
pytorch-fid
h5py
lpips
tabulate
ninja
matplotlib
webdataset
braceexpand
Pillow
accelerate==0.29.3
compel==0.1.8
datasets
filelock
flax>=0.4.1
hf-doc-builder>=0.3.0
huggingface-hub>=0.20.2
requests-mock==1.10.0
importlib_metadata
invisible-watermark>=0.2.0
isort>=5.5.4
jax>=0.4.1
jaxlib>=0.4.1
Jinja2
k-diffusion>=0.0.12
torchsde
note_seq
librosa
numpy
parameterized
git+https://github.com/yhZhai/peft.git
protobuf==3.20.3
pytest
pytest-timeout
pytest-xdist
ruff==0.1.5
safetensors>=0.3.1
sentencepiece>=0.1.91,!=0.1.92
GitPython<3.1.19
# scipy==1.11.1
onnx
regex!=2019.12.17
requests
bitsandbytes
git+https://github.com/microsoft/azfuse.git
deepspeed==0.11.2
# deepspeed==0.6.6
albumentations
mlflow
moviepy
git+https://github.com/openai/CLIP.git
av
git+https://github.com/yhZhai/open_clip.git