torch==2.3.1
torchvision==0.18.1
gradio==4.37.2
diffusers==0.31.0
transformers==4.45.2
accelerate