torch>=2.2
diffusers>=0.24.0
transformers
tqdm
