diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-10 11:55:09 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-10 11:55:09 +0300 |
| commit | d25219b7e889cf34bccae9cb88497708796efda2 (patch) | |
| tree | 0fe0a3860a440b08ad9ec48b689bc1cb1016bac5 /modules/models/diffusion/ddpm_edit.py | |
| parent | a5121e7a0623db328a9462d340d389ed6737374a (diff) | |
manual fixes for some C408
Diffstat (limited to 'modules/models/diffusion/ddpm_edit.py')
| -rw-r--r-- | modules/models/diffusion/ddpm_edit.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/modules/models/diffusion/ddpm_edit.py b/modules/models/diffusion/ddpm_edit.py index af4dea15..3fb76b65 100644 --- a/modules/models/diffusion/ddpm_edit.py +++ b/modules/models/diffusion/ddpm_edit.py @@ -405,7 +405,7 @@ class DDPM(pl.LightningModule): @torch.no_grad() def log_images(self, batch, N=8, n_row=2, sample=True, return_keys=None, **kwargs): - log = dict() + log = {} x = self.get_input(batch, self.first_stage_key) N = min(x.shape[0], N) n_row = min(x.shape[0], n_row) @@ -413,7 +413,7 @@ class DDPM(pl.LightningModule): log["inputs"] = x # get diffusion row - diffusion_row = list() + diffusion_row = [] x_start = x[:n_row] for t in range(self.num_timesteps): @@ -1263,7 +1263,7 @@ class LatentDiffusion(DDPM): use_ddim = False - log = dict() + log = {} z, c, x, xrec, xc = self.get_input(batch, self.first_stage_key, return_first_stage_outputs=True, force_c_encode=True, @@ -1291,7 +1291,7 @@ class LatentDiffusion(DDPM): if plot_diffusion_rows: # get diffusion row - diffusion_row = list() + diffusion_row = [] z_start = z[:n_row] for t in range(self.num_timesteps): if t % self.log_every_t == 0 or t == self.num_timesteps - 1: |