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