accelerate >= 1.2.1
diffusers >= 0.31.0
transformers >= 4.47.0
jax >= 0.4.28