diff options
| author | Aarni Koskela <akx@iki.fi> | 2023-06-02 11:58:10 +0000 |
|---|---|---|
| committer | Aarni Koskela <akx@iki.fi> | 2023-06-02 12:07:10 +0000 |
| commit | 51864790fd72386fbbbb015d24a43ce501ecaa4b (patch) | |
| tree | a114dc065c25ec267043f35dea735f8e3097e9a0 /extensions-builtin/LDSR/sd_hijack_autoencoder.py | |
| parent | 3e995778fc525ff15d56c1472c1a1bc701019ec5 (diff) | |
| download | stable-diffusion-webui-gfx803-51864790fd72386fbbbb015d24a43ce501ecaa4b.tar.gz stable-diffusion-webui-gfx803-51864790fd72386fbbbb015d24a43ce501ecaa4b.tar.bz2 stable-diffusion-webui-gfx803-51864790fd72386fbbbb015d24a43ce501ecaa4b.zip | |
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): |