# --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