aboutsummaryrefslogtreecommitdiff
path: root/extensions-builtin/LDSR/sd_hijack_autoencoder.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-05-10 11:05:02 +0300
committerAUTOMATIC <16777216c@gmail.com>2023-05-10 11:05:02 +0300
commit028d3f6425d85f122027c127fba8bcbf4f66ee75 (patch)
tree05e714f118aaa4d25920918807e0b83344133629 /extensions-builtin/LDSR/sd_hijack_autoencoder.py
parente42de4b8a2356c6d286adb07292442d75e5595d3 (diff)
ruff auto fixes
Diffstat (limited to 'extensions-builtin/LDSR/sd_hijack_autoencoder.py')
-rw-r--r--extensions-builtin/LDSR/sd_hijack_autoencoder.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/extensions-builtin/LDSR/sd_hijack_autoencoder.py b/extensions-builtin/LDSR/sd_hijack_autoencoder.py
index 6303fed5..f457ca93 100644
--- a/extensions-builtin/LDSR/sd_hijack_autoencoder.py
+++ b/extensions-builtin/LDSR/sd_hijack_autoencoder.py
@@ -288,5 +288,5 @@ class VQModelInterface(VQModel):
dec = self.decoder(quant)
return dec
-setattr(ldm.models.autoencoder, "VQModel", VQModel)
-setattr(ldm.models.autoencoder, "VQModelInterface", VQModelInterface)
+ldm.models.autoencoder.VQModel = VQModel
+ldm.models.autoencoder.VQModelInterface = VQModelInterface