accelerate >= 1.2.1
diffusers >= 0.31.0
transformers >= 4.49.0
jax >= 0.4.28
timm >= 1.0.15