rich-text-to-image / requirements.txt
songweig's picture
fix req
2ce222a
--extra-index-url https://download.pytorch.org/whl/cu117
torch==1.13.1
torchvision==0.14.1
diffusers==0.18.2
transformers==4.27.0
safetensors==0.3.1
invisible_watermark==0.2.0
numpy==1.24.3
seaborn==0.12.2
accelerate==0.16.0
scikit-learn==1.1.3