diff options
| author | Aarni Koskela <akx@iki.fi> | 2023-05-09 22:17:58 +0300 |
|---|---|---|
| committer | Aarni Koskela <akx@iki.fi> | 2023-05-09 22:25:39 +0300 |
| commit | 3ba6c3c83c0983a025c7bddc08bb7f49481b3cbb (patch) | |
| tree | 77b2bea31c0e94ac67fddf86fdc7cba65a24af2c /modules/models/diffusion/ddpm_edit.py | |
| parent | 8fb16ceb288a93f6ecaa45a0afcb3db718333e3e (diff) | |
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) |