torch
dacite
shortuuid
huggingface_hub[cli]

[dev]
ipdb
jupyterlab
ipython
black
isort

[train]
omegaconf
wandb
tensorboard
tensorboardX
diffusers
msgpack
torchvision
