diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-06-09 19:48:18 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-06-09 19:48:18 +0000 |
| commit | 3b11f17a374520e493e120e7f47443acd97393c8 (patch) | |
| tree | 743d96990f646fb49ff377a67849c288c31e177e /extensions-builtin/LDSR/sd_hijack_ddpm_v1.py | |
| parent | baf6946e06249c5af9851c60171692c44ef633e0 (diff) | |
| parent | 59419bd64a1581caccaac04dceb66c1c069a2db1 (diff) | |
| download | stable-diffusion-webui-gfx803-3b11f17a374520e493e120e7f47443acd97393c8.tar.gz stable-diffusion-webui-gfx803-3b11f17a374520e493e120e7f47443acd97393c8.tar.bz2 stable-diffusion-webui-gfx803-3b11f17a374520e493e120e7f47443acd97393c8.zip | |
Merge branch 'dev' into release_candidate
Diffstat (limited to 'extensions-builtin/LDSR/sd_hijack_ddpm_v1.py')
| -rw-r--r-- | extensions-builtin/LDSR/sd_hijack_ddpm_v1.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/extensions-builtin/LDSR/sd_hijack_ddpm_v1.py b/extensions-builtin/LDSR/sd_hijack_ddpm_v1.py index 631a08ef..04adc5eb 100644 --- a/extensions-builtin/LDSR/sd_hijack_ddpm_v1.py +++ b/extensions-builtin/LDSR/sd_hijack_ddpm_v1.py @@ -195,9 +195,9 @@ class DDPMV1(pl.LightningModule): missing, unexpected = self.load_state_dict(sd, strict=False) if not only_model else self.model.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 len(unexpected) > 0: + if unexpected: print(f"Unexpected Keys: {unexpected}") def q_mean_variance(self, x_start, t): |