requirements.txt

# These requirements are for GPU
torch==2.1.0+cu118
torchvision==0.16.0+cu118
transformers==4.34.1
diffusers==0.21.4
gradio==3.50.2
ftfy==6.1.1
accelerate==0.23.0