torch==2.3.1
torchsde
torchvision==0.18.1
torchaudio==2.3.1
einops
transformers>=4.28.1
tokenizers>=0.13.3
sentencepiece
safetensors>=0.4.2
aiohttp
pyyaml
Pillow
scipy
tqdm
psutil
numpy<2.0.0

#non essential dependencies:
kornia>=0.7.1
spandrel
soundfile