torch==2.1.0
diffusers
transformers
gradio==4.21.0
peft
imageio
invisible-watermark
omegaconf