diff options
| author | Aarni Koskela <akx@iki.fi> | 2023-05-11 15:28:15 +0000 |
|---|---|---|
| committer | Aarni Koskela <akx@iki.fi> | 2023-05-11 17:29:11 +0000 |
| commit | 49a55b410b66b7dd9be9335d8a2e3a71e4f8b15c (patch) | |
| tree | d79f004eae46bc1c49832f3c668a524107c30034 /extensions-builtin/LDSR/sd_hijack_ddpm_v1.py | |
| parent | 431bc5a297ff7c17231b92b6c8f8152b2fab8553 (diff) | |
| download | stable-diffusion-webui-gfx803-49a55b410b66b7dd9be9335d8a2e3a71e4f8b15c.tar.gz stable-diffusion-webui-gfx803-49a55b410b66b7dd9be9335d8a2e3a71e4f8b15c.tar.bz2 stable-diffusion-webui-gfx803-49a55b410b66b7dd9be9335d8a2e3a71e4f8b15c.zip | |
Autofix Ruff W (not W605) (mostly whitespace)
Diffstat (limited to 'extensions-builtin/LDSR/sd_hijack_ddpm_v1.py')
| -rw-r--r-- | extensions-builtin/LDSR/sd_hijack_ddpm_v1.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/extensions-builtin/LDSR/sd_hijack_ddpm_v1.py b/extensions-builtin/LDSR/sd_hijack_ddpm_v1.py index 57c02d12..631a08ef 100644 --- a/extensions-builtin/LDSR/sd_hijack_ddpm_v1.py +++ b/extensions-builtin/LDSR/sd_hijack_ddpm_v1.py @@ -460,7 +460,7 @@ class LatentDiffusionV1(DDPMV1): self.instantiate_cond_stage(cond_stage_config) self.cond_stage_forward = cond_stage_forward self.clip_denoised = False - self.bbox_tokenizer = None + self.bbox_tokenizer = None self.restarted_from_ckpt = False if ckpt_path is not None: @@ -792,7 +792,7 @@ class LatentDiffusionV1(DDPMV1): z = z.view((z.shape[0], -1, ks[0], ks[1], z.shape[-1])) # (bn, nc, ks[0], ks[1], L ) # 2. apply model loop over last dim - if isinstance(self.first_stage_model, VQModelInterface): + if isinstance(self.first_stage_model, VQModelInterface): output_list = [self.first_stage_model.decode(z[:, :, :, :, i], force_not_quantize=predict_cids or force_not_quantize) for i in range(z.shape[-1])] @@ -890,7 +890,7 @@ class LatentDiffusionV1(DDPMV1): if hasattr(self, "split_input_params"): assert len(cond) == 1 # todo can only deal with one conditioning atm - assert not return_ids + assert not return_ids ks = self.split_input_params["ks"] # eg. (128, 128) stride = self.split_input_params["stride"] # eg. (64, 64) |