# Essential dependencies
torch
torchsde
torchvision
torchaudio
einops
transformers>=4.28.1
tokenizers>=0.13.3
sentencepiece
safetensors>=0.4.2
aiohttp
pyyaml
Pillow
scipy
tqdm
psutil

# Non-essential dependencies
kornia>=0.7.1
spandrel
soundfile
imageio-ffmpeg
ffmpeg-python
opencv-python
onnxruntime-gpu
pykalman
mediapipe
onnx2torch
insightface

numpy<=1.26.4

psycopg2-binary
boto3

psycopg2-binary
firebase_admin