requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu118
torch
torchvision
torchaudio
timm
diffusers[torch]
cmake
ninja
accelerate
tensorboard
pillow
einops
transformers
av
scikit-image
decord
pandas
imageio
imageio-ffmpeg
sentencepiece
beautifulsoup4
ftfy
omegaconf
gradio
imageio
imageio-ffmpeg
bitsandbytes
xformers
setuptools
pip
wheel
triton
spaces
huggingface-hub
numpy
matplotlib
lit
pybind11