torch == 2.4.0
torchvision
torchaudio
python-dotenv
cython
gitpython >= 3.1
tensorboard >= 2.11
numpy >= 1.21, <2.1
Pillow >= 9.5
opencv-python >= 4.8
scipy >= 1.7
tqdm >= 4.66.1
gradio >= 3.34
einops >= 0.6
hydra-core >= 1.3.2
requests
torchdiffeq
librosa >= 0.8.1
nitrous-ema
safetensors
auraloss
hydra_colorlog
tensordict
colorlog
open_clip_torch
soundfile
av
replicate

accelerate
diffusers==0.30.0
invisible_watermark
transformers==4.43.3
xformers
sentencepiece
peft
gradio
opencv-python-headless
numpy
sacremoses