numpy<2
diffusers==0.18.2 #  20
# xformers==0.0.21  # optional
transformers==4.30.1
einops
pytorch_lightning
omegaconf
matplotlib
spaces  # optional unless gradio is required
gradio  # optional unless gradio is required
huggingface-hub==0.25.1
librosa
opencv-python
progressbar
pydantic==2.10.6
fastapi==0.112.2 # temporary fix following /static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fnew-gradio-space-connection-errored-out%2F105509%2F3
# wandb
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.1.0
torchaudio==2.1.0
torchvision==0.16.0