| numpy | |
| tqdm | |
| psutil | |
| packaging | |
| pre-commit | |
| rich | |
| click | |
| fabric | |
| contexttimer | |
| ninja | |
| torch==2.3.0 | |
| safetensors | |
| einops | |
| pydantic | |
| ray | |
| protobuf | |
| gdown | |
| pyav | |
| tensorboard | |
| timm | |
| matplotlib | |
| accelerate | |
| diffusers | |
| transformers | |
| colossalai | |
| omegaconf | |
| xformers==0.0.26.post1 | |
| ftfy | |
| rotary_embedding_torch | |
| imageio | |
| imageio-ffmpeg | |
| SwissArmyTransformer==0.4.12 | |
| wandb==0.17.5 | |
| pytorch_lightning==2.3.3 | |
| kornia==0.7.3 | |
| beartype==0.18.5 | |