Add missing reqs / fix versions

2022-11-04 17:31:08 -07:00 · 2022-11-04 17:31:08 -07:00 · bb153c1ec0
parent 6b73cfe448
commit bb153c1ec0
1 changed files with 4 additions and 2 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -11,7 +11,8 @@ streamlit>=0.73.1
 einops==0.3.0
 torch-fidelity==0.3.0
 transformers==4.19.2
-torchmetrics==0.6.0
+diffusers==0.7.1
+torchmetrics==0.7.0
 kornia==0.6
 gradio
 git+https://github.com/illeatmyhat/taming-transformers.git@master#egg=taming-transformers
@ -19,4 +20,5 @@ git+https://github.com/openai/CLIP.git@main#egg=clip
 git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion
 webdataset
 wandb
-fairscale
+fairscale
+pynvml==11.4.1