peal-xai
omegaconf>=2.3
einops>=0.6
safetensors>=0.4
timm>=0.9
diffusers>=0.28
transformers>=4.36
huggingface-hub>=0.23

[train]
datasets>=2.14
lpips>=0.1.4
torch-fidelity>=0.3
wandb>=0.15
webdataset>=0.2
gram-newton-schulz
