aboutsummaryrefslogtreecommitdiff
path: root/extensions-builtin/LDSR/sd_hijack_autoencoder.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-05-31 18:46:15 +0300
committerGitHub <noreply@github.com>2023-05-31 18:46:15 +0300
commit0b0f60f9549f067791418c085ac2a306a147c8d3 (patch)
treefd1b3c1095d7b0fc5675e3feb4754a59e2cec044 /extensions-builtin/LDSR/sd_hijack_autoencoder.py
parent69f49a935a13d9e89af7404de1bef617dfaf9285 (diff)
parent5fcdaa6a7f19d083a6393cc0d2b933ff5080f5b3 (diff)
Merge pull request #10856 from akx/untamed
Remove taming_transformers dependency
Diffstat (limited to 'extensions-builtin/LDSR/sd_hijack_autoencoder.py')
-rw-r--r--extensions-builtin/LDSR/sd_hijack_autoencoder.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/extensions-builtin/LDSR/sd_hijack_autoencoder.py b/extensions-builtin/LDSR/sd_hijack_autoencoder.py
index 81c5101b..27a86e13 100644
--- a/extensions-builtin/LDSR/sd_hijack_autoencoder.py
+++ b/extensions-builtin/LDSR/sd_hijack_autoencoder.py
@@ -10,7 +10,7 @@ from contextlib import contextmanager
from torch.optim.lr_scheduler import LambdaLR
from ldm.modules.ema import LitEma
-from taming.modules.vqvae.quantize import VectorQuantizer2 as VectorQuantizer
+from vqvae_quantize import VectorQuantizer2 as VectorQuantizer
from ldm.modules.diffusionmodules.model import Encoder, Decoder
from ldm.util import instantiate_from_config