rich-text-to-image / requirements.txt
songweig's picture
update enviroment
70acd79
raw history blame
No virus
199 Bytes
--extra-index-url https://download.pytorch.org/whl/cu117
torch==1.13.1
torchvision==0.14.1
diffusers==0.12.1
transformers==4.26.0
numpy==1.24.2
seaborn==0.12.2
accelerate==0.16.0
scikit-learn==0.24.1