| # 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 |