diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-05-09 19:40:51 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-05-09 19:40:51 +0000 |
| commit | d6a9b22c195a0b4b9c6b04ccdfc59c18d099da7a (patch) | |
| tree | 894f00b92caa229dbfaa2d07c16c25e0167412db /modules/models/diffusion/ddpm_edit.py | |
| parent | ccbb361845bfabfa3fc2c77bb234ea4be0781dd9 (diff) | |
| parent | 3ba6c3c83c0983a025c7bddc08bb7f49481b3cbb (diff) | |
| download | stable-diffusion-webui-gfx803-d6a9b22c195a0b4b9c6b04ccdfc59c18d099da7a.tar.gz stable-diffusion-webui-gfx803-d6a9b22c195a0b4b9c6b04ccdfc59c18d099da7a.tar.bz2 stable-diffusion-webui-gfx803-d6a9b22c195a0b4b9c6b04ccdfc59c18d099da7a.zip | |
Merge pull request #10232 from akx/eff
Fix up string formatting/concatenation to f-strings where feasible
Diffstat (limited to 'modules/models/diffusion/ddpm_edit.py')
| -rw-r--r-- | modules/models/diffusion/ddpm_edit.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/models/diffusion/ddpm_edit.py b/modules/models/diffusion/ddpm_edit.py index f3d49c44..f880bc3c 100644 --- a/modules/models/diffusion/ddpm_edit.py +++ b/modules/models/diffusion/ddpm_edit.py @@ -223,7 +223,7 @@ class DDPM(pl.LightningModule): for k in keys: for ik in ignore_keys: if k.startswith(ik): - print("Deleting key {} from state_dict.".format(k)) + print(f"Deleting key {k} from state_dict.") del sd[k] missing, unexpected = self.load_state_dict(sd, strict=False) if not only_model else self.model.load_state_dict( sd, strict=False) @@ -386,7 +386,7 @@ class DDPM(pl.LightningModule): _, loss_dict_no_ema = self.shared_step(batch) with self.ema_scope(): _, loss_dict_ema = self.shared_step(batch) - loss_dict_ema = {key + '_ema': loss_dict_ema[key] for key in loss_dict_ema} + loss_dict_ema = {f"{key}_ema": loss_dict_ema[key] for key in loss_dict_ema} self.log_dict(loss_dict_no_ema, prog_bar=False, logger=True, on_step=False, on_epoch=True) self.log_dict(loss_dict_ema, prog_bar=False, logger=True, on_step=False, on_epoch=True) |