| numpy==1.26.4 | |
| sympy>=1.10 | |
| accelerate | |
| datasets | |
| diffusers[torch] | |
| transformers | |
| safetensors | |
| smart_open | |
| dotwiz | |
| omegaconf | |
| tensorboard | |
| huggingface-hub | |
| einops | |
| lpips | |
| timm | |
| scipy | |
| scikit-learn | |
| scikit-image | |
| kornia | |
| torchtyping | |
| git+https://github.com/xwen99/torch-fidelity.git@master#egg=torch-fidelity | |
| open_clip_torch | |
| opencv-python-headless | |
| torchmetrics | |
| torchdiffeq | |
| lmdb | |
| triton==3.0.0 | |
| matplotlib |