--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.3.1+cu121
torchvision==0.18.1+cu121

accelerate==0.31.0
diffusers==0.26.0
gradio==4.37.2
sentencepiece==0.2.0
transformers==4.42.3