aboutsummaryrefslogtreecommitdiffstats
path: root/modules/ldsr_model_arch.py
diff options
context:
space:
mode:
authord8ahazard <d8ahazard@gmail.com>2022-09-30 13:55:04 +0000
committerd8ahazard <d8ahazard@gmail.com>2022-09-30 13:55:04 +0000
commit8d60645106d7e2daa0da89c5b21d7ffdac61cf9e (patch)
treeedb87c4e2a86cb5b2a9abeedd0543391d1982443 /modules/ldsr_model_arch.py
parent64c6b13312ff3a20f48781c4c3780355c4b7b2af (diff)
downloadstable-diffusion-webui-gfx803-8d60645106d7e2daa0da89c5b21d7ffdac61cf9e.tar.gz
stable-diffusion-webui-gfx803-8d60645106d7e2daa0da89c5b21d7ffdac61cf9e.tar.bz2
stable-diffusion-webui-gfx803-8d60645106d7e2daa0da89c5b21d7ffdac61cf9e.zip
Fix model paths, ensure we have the right files.
Also, clean up logging in the ldsr arch file.
Diffstat (limited to 'modules/ldsr_model_arch.py')
-rw-r--r--modules/ldsr_model_arch.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/modules/ldsr_model_arch.py b/modules/ldsr_model_arch.py
index 7faac6e1..093a3210 100644
--- a/modules/ldsr_model_arch.py
+++ b/modules/ldsr_model_arch.py
@@ -100,7 +100,6 @@ class LDSR:
# 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 +110,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"]