diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-05-31 19:31:19 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-05-31 19:31:19 +0300 |
| commit | 177d4b6828eee6a9aa1e9f2cf4877ad9ae8fcae4 (patch) | |
| tree | 466c92812f7b978cf3a535187c18be78c4e26d37 /extensions-builtin/LDSR/sd_hijack_autoencoder.py | |
| parent | c1a5068ebea127412dfaaa6598795196a64200f1 (diff) | |
| parent | 881de0df38c1fa6d0d61f7bc6fc93c100a9f35d0 (diff) | |
Merge branch 'dev' into sync-req
Diffstat (limited to 'extensions-builtin/LDSR/sd_hijack_autoencoder.py')
| -rw-r--r-- | extensions-builtin/LDSR/sd_hijack_autoencoder.py | 2 |
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 |