diff --git a/requirements.txt b/requirements.txt index c9e3f2fc..cb27daf5 100644 --- 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 diff --git a/requirements_versions.txt b/requirements_versions.txt index d2277b81..754b163c 100644 --- 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