aboutsummaryrefslogtreecommitdiffstats
path: root/extensions-builtin/LDSR/ldsr_model_arch.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-05-11 18:25:15 +0000
committerGitHub <noreply@github.com>2023-05-11 18:25:15 +0000
commitabe32cefa39dee36d7f661d4e63c28ea8dd60c4f (patch)
tree1f1d817b59b49c6d3944c959151ce4c67d9041da /extensions-builtin/LDSR/ldsr_model_arch.py
parentb4aaa339d529c81859858f0bedcc72b44fccd3d0 (diff)
parent49a55b410b66b7dd9be9335d8a2e3a71e4f8b15c (diff)
downloadstable-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/ldsr_model_arch.py')
-rw-r--r--extensions-builtin/LDSR/ldsr_model_arch.py4
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"]