accelerate
adam-atan2-pytorch>=0.0.12
beartype
classifier-free-guidance-pytorch>=0.6.10
einops>=0.8.0
ema-pytorch>=0.5.3
jaxtyping
numpy
sentencepiece
torch>=2.0
