transformers==4.46.0
wandb
webdataset
omegaconf
pyarrow
pandas
jaxtyping
diffusers==0.32.2
typeguard
accelerate
deepspeed
lightning
datasets
torchvision