aboutsummaryrefslogtreecommitdiffstats
path: root/modules/ldsr_model_arch.py
diff options
context:
space:
mode:
authord8ahazard <d8ahazard@gmail.com>2022-09-30 00:59:53 +0000
committerd8ahazard <d8ahazard@gmail.com>2022-09-30 00:59:53 +0000
commit435fd2112aee9a0e61408ac56663e41beea1e446 (patch)
tree52ba152eb04e1c51f17cfbd6091b1b76b696fd7e /modules/ldsr_model_arch.py
parentd73741794d38a5c1aacacc7a6ed3fe3ca65724db (diff)
downloadstable-diffusion-webui-gfx803-435fd2112aee9a0e61408ac56663e41beea1e446.tar.gz
stable-diffusion-webui-gfx803-435fd2112aee9a0e61408ac56663e41beea1e446.tar.bz2
stable-diffusion-webui-gfx803-435fd2112aee9a0e61408ac56663e41beea1e446.zip
Fixes, cleanup.
Diffstat (limited to 'modules/ldsr_model_arch.py')
-rw-r--r--modules/ldsr_model_arch.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/modules/ldsr_model_arch.py b/modules/ldsr_model_arch.py
index 8fe87c6a..f8f3c3d3 100644
--- a/modules/ldsr_model_arch.py
+++ b/modules/ldsr_model_arch.py
@@ -101,8 +101,10 @@ class LDSR:
print("Foo")
down_sample_rate = target_scale / 4
print(f"Downsample rate is {down_sample_rate}")
- width_downsampled_pre = width_og * down_sample_rate
- height_downsampled_pre = height_og * down_sample_method
+ wd = width_og * down_sample_rate
+ hd = height_og * down_sample_rate
+ width_downsampled_pre = int(wd)
+ height_downsampled_pre = int(hd)
if down_sample_rate != 1:
print(