This commit is contained in:
Yuwei Guo
2023-07-09 21:32:22 +08:00
parent 260cc266cd
commit e2590df101
21 changed files with 4414 additions and 0 deletions

9
requirements.txt Normal file
View File

@@ -0,0 +1,9 @@
--extra-index-url https://download.pytorch.org/whl/cu113
torch==1.12.1+cu113
torchvision==0.13.1+cu113
diffusers[torch]==0.11.1
transformers==4.25.1
einops
omegaconf
safetensors
imageio==2.27.0