torch
torchvision
einops
transformers
diffusers
huggingface_hub

[demo]
gradio

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