aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_hijack_autoencoder.py
diff options
context:
space:
mode:
authorwywywywy <wywywywy@gmail.com>2022-11-29 19:22:53 +0000
committerGitHub <noreply@github.com>2022-11-29 19:22:53 +0000
commit7193814cf7b052fe52596cab89c5f0fd95823950 (patch)
treeafbbc7272f6f8cae76f42db73fef84fc18257bff /modules/sd_hijack_autoencoder.py
parent36c3613d16c523e43ec4dedbcbe9a3b93ad7d139 (diff)
downloadstable-diffusion-webui-gfx803-7193814cf7b052fe52596cab89c5f0fd95823950.tar.gz
stable-diffusion-webui-gfx803-7193814cf7b052fe52596cab89c5f0fd95823950.tar.bz2
stable-diffusion-webui-gfx803-7193814cf7b052fe52596cab89c5f0fd95823950.zip
Added purpose of this hijack to comments
Diffstat (limited to 'modules/sd_hijack_autoencoder.py')
-rw-r--r--modules/sd_hijack_autoencoder.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/modules/sd_hijack_autoencoder.py b/modules/sd_hijack_autoencoder.py
index ffa72f90..8e03c7f8 100644
--- a/modules/sd_hijack_autoencoder.py
+++ b/modules/sd_hijack_autoencoder.py
@@ -1,3 +1,7 @@
+# The content of this file comes from the ldm/models/autoencoder.py file of the compvis/stable-diffusion repo
+# The VQModel & VQModelInterface were subsequently removed from ldm/models/autoencoder.py when we moved to the stability-ai/stablediffusion repo
+# As the LDSR upscaler relies on VQModel & VQModelInterface, the hijack aims to put them back into the ldm.models.autoencoder
+
import torch
import pytorch_lightning as pl
import torch.nn.functional as F