lightning==2.5.0.post0
omegaconf==2.3.0
numpy==1.26.0
torch==2.2.0
diffusers==0.30.0
jsonargparse[signatures]>=4.27.7
accelerate