added dependency for gfpgan

3 years ago · 0bfa0d4381
parent c6e7c280fc
commit 0bfa0d4381
2 changed files with 2 additions and 0 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -8,6 +8,7 @@ torch
 transformers
 omegaconf
 pytorch_lightning
+basicsr
 diffusers
 invisible-watermark
 git+https://github.com/crowsonkb/k-diffusion.git
--- a/requirements_versions.txt
+++ b/requirements_versions.txt
@ -8,3 +8,4 @@ torch
 transformers==4.19.2
 omegaconf==2.1.1
 pytorch_lightning==1.7.2
+basicsr==1.3.5