aboutsummaryrefslogtreecommitdiffstats
path: root/modules/ldsr_model_arch.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2022-09-30 20:09:18 +0000
committerGitHub <noreply@github.com>2022-09-30 20:09:18 +0000
commitf253790b40a71f2d88951c26224e0c80b6baea52 (patch)
tree786b67f0cde4c15b3a0112e02495a51978ea9d26 /modules/ldsr_model_arch.py
parent3f64e23f81618395428da0a132bf3a6cfd5c42c9 (diff)
parenta9a0320d2688a292fe834986c9abe141ff29bf7c (diff)
downloadstable-diffusion-webui-gfx803-f253790b40a71f2d88951c26224e0c80b6baea52.tar.gz
stable-diffusion-webui-gfx803-f253790b40a71f2d88951c26224e0c80b6baea52.tar.bz2
stable-diffusion-webui-gfx803-f253790b40a71f2d88951c26224e0c80b6baea52.zip
Merge pull request #1371 from d8ahazard/master
Fix LDSR, BSRGAN, recursive SD checkpoint Loading
Diffstat (limited to 'modules/ldsr_model_arch.py')
-rw-r--r--modules/ldsr_model_arch.py4
1 files changed, 1 insertions, 3 deletions
diff --git a/modules/ldsr_model_arch.py b/modules/ldsr_model_arch.py
index 7faac6e1..14db5076 100644
--- a/modules/ldsr_model_arch.py
+++ b/modules/ldsr_model_arch.py
@@ -98,9 +98,7 @@ class LDSR:
im_og = image
width_og, height_og = im_og.size
# If we can adjust the max upscale size, then the 4 below should be our variable
- print("Foo")
down_sample_rate = target_scale / 4
- print(f"Downsample rate is {down_sample_rate}")
wd = width_og * down_sample_rate
hd = height_og * down_sample_rate
width_downsampled_pre = int(wd)
@@ -111,7 +109,7 @@ class LDSR:
f'Downsampling from [{width_og}, {height_og}] to [{width_downsampled_pre}, {height_downsampled_pre}]')
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")
+ print(f"Down sample rate is 1 from {target_scale} / 4 (Not downsampling)")
logs = self.run(model["model"], im_og, diffusion_steps, eta)
sample = logs["sample"]