torch
torchvision
einops
transformers
diffusers
huggingface_hub>=0.23

[demo]
gradio

[train]
hydra-core
lightning
torch-fidelity
pandas
wandb
timm
scikit-learn
webdataset==0.2.57
lovely_tensors
