diff options
-rw-r--r-- | requirements.txt | 1 | ||||
-rw-r--r-- | requirements_versions.txt | 1 |
2 files changed, 2 insertions, 0 deletions
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
|