diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-06-27 06:02:38 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-06-27 06:02:38 +0000 |
| commit | 58a9a261c4feb729717c386fea70f8ea985c6722 (patch) | |
| tree | ebc822c17a889034672297d7450477f7c945cbb7 /extensions-builtin/LDSR/sd_hijack_autoencoder.py | |
| parent | 373ff5a217eca33607abb692b9ebfa38abb7fe33 (diff) | |
| parent | 2c43dd766da52d2ccaaa78d8f265a6a4aa33b9df (diff) | |
| download | stable-diffusion-webui-gfx803-58a9a261c4feb729717c386fea70f8ea985c6722.tar.gz stable-diffusion-webui-gfx803-58a9a261c4feb729717c386fea70f8ea985c6722.tar.bz2 stable-diffusion-webui-gfx803-58a9a261c4feb729717c386fea70f8ea985c6722.zip | |
Merge branch 'dev' into meta_class
Diffstat (limited to 'extensions-builtin/LDSR/sd_hijack_autoencoder.py')
| -rw-r--r-- | extensions-builtin/LDSR/sd_hijack_autoencoder.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/extensions-builtin/LDSR/sd_hijack_autoencoder.py b/extensions-builtin/LDSR/sd_hijack_autoencoder.py index 81c5101b..c29d274d 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 @@ -91,8 +91,9 @@ class VQModel(pl.LightningModule): del sd[k] missing, unexpected = self.load_state_dict(sd, strict=False) print(f"Restored from {path} with {len(missing)} missing and {len(unexpected)} unexpected keys") - if len(missing) > 0: + if missing: print(f"Missing Keys: {missing}") + if unexpected: print(f"Unexpected Keys: {unexpected}") def on_train_batch_end(self, *args, **kwargs): |