diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-06-05 08:06:37 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-06-05 08:06:37 +0000 |
| commit | 7a7a201d819ea9920a1f067c556ec3998911f01a (patch) | |
| tree | 5a834df58178642477e0ab8bd0ab0a53e66f153a /extensions-builtin/LDSR/sd_hijack_autoencoder.py | |
| parent | 9781f31f74c1f6acf8e1260f076aeb7a71e2766f (diff) | |
| parent | 51864790fd72386fbbbb015d24a43ce501ecaa4b (diff) | |
| download | stable-diffusion-webui-gfx803-7a7a201d819ea9920a1f067c556ec3998911f01a.tar.gz stable-diffusion-webui-gfx803-7a7a201d819ea9920a1f067c556ec3998911f01a.tar.bz2 stable-diffusion-webui-gfx803-7a7a201d819ea9920a1f067c556ec3998911f01a.zip | |
Merge pull request #10956 from akx/len
Simplify a bunch of `len(x) > 0`/`len(x) == 0` style expressions
Diffstat (limited to 'extensions-builtin/LDSR/sd_hijack_autoencoder.py')
| -rw-r--r-- | extensions-builtin/LDSR/sd_hijack_autoencoder.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/extensions-builtin/LDSR/sd_hijack_autoencoder.py b/extensions-builtin/LDSR/sd_hijack_autoencoder.py index 27a86e13..c29d274d 100644 --- a/extensions-builtin/LDSR/sd_hijack_autoencoder.py +++ b/extensions-builtin/LDSR/sd_hijack_autoencoder.py @@ -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): |