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