diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-05-11 18:25:15 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-05-11 18:25:15 +0000 |
| commit | abe32cefa39dee36d7f661d4e63c28ea8dd60c4f (patch) | |
| tree | 1f1d817b59b49c6d3944c959151ce4c67d9041da /extensions-builtin/LDSR/sd_hijack_ddpm_v1.py | |
| parent | b4aaa339d529c81859858f0bedcc72b44fccd3d0 (diff) | |
| parent | 49a55b410b66b7dd9be9335d8a2e3a71e4f8b15c (diff) | |
| download | stable-diffusion-webui-gfx803-abe32cefa39dee36d7f661d4e63c28ea8dd60c4f.tar.gz stable-diffusion-webui-gfx803-abe32cefa39dee36d7f661d4e63c28ea8dd60c4f.tar.bz2 stable-diffusion-webui-gfx803-abe32cefa39dee36d7f661d4e63c28ea8dd60c4f.zip | |
Merge pull request #10285 from akx/ruff-spacing
Indentation + ruff whitespace fixes
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) |