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/ldsr_model_arch.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/ldsr_model_arch.py')
| -rw-r--r-- | extensions-builtin/LDSR/ldsr_model_arch.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/extensions-builtin/LDSR/ldsr_model_arch.py b/extensions-builtin/LDSR/ldsr_model_arch.py index 2173de79..7f450086 100644 --- a/extensions-builtin/LDSR/ldsr_model_arch.py +++ b/extensions-builtin/LDSR/ldsr_model_arch.py @@ -130,11 +130,11 @@ class LDSR: im_og = im_og.resize((width_downsampled_pre, height_downsampled_pre), Image.LANCZOS) else: print(f"Down sample rate is 1 from {target_scale} / 4 (Not downsampling)") - + # pad width and height to multiples of 64, pads with the edge values of image to avoid artifacts pad_w, pad_h = np.max(((2, 2), np.ceil(np.array(im_og.size) / 64).astype(int)), axis=0) * 64 - im_og.size im_padded = Image.fromarray(np.pad(np.array(im_og), ((0, pad_h), (0, pad_w), (0, 0)), mode='edge')) - + logs = self.run(model["model"], im_padded, diffusion_steps, eta) sample = logs["sample"] |