Mirrors
/
waifu-diffusion
mirror of https://github.com/harubaru/waifu-diffusion.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add missing reqs / fix versions

This commit is contained in:
Astralite Heart 2022-11-04 17:31:08 -07:00
parent 6b73cfe448
commit bb153c1ec0
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
requirements.txt
View File

@ -11,7 +11,8 @@ streamlit>=0.73.1
einops==0.3.0 einops==0.3.0
torch-fidelity==0.3.0 torch-fidelity==0.3.0
transformers==4.19.2 transformers==4.19.2
torchmetrics==0.6.0 diffusers==0.7.1
torchmetrics==0.7.0
kornia==0.6 kornia==0.6
gradio gradio
git+https://github.com/illeatmyhat/taming-transformers.git@master#egg=taming-transformers git+https://github.com/illeatmyhat/taming-transformers.git@master#egg=taming-transformers
@ -20,3 +21,4 @@ git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion
webdataset webdataset
wandb wandb
fairscale fairscale
pynvml==11.4.1