Merge branch 'dev' into torch210

1 files changed, 12 insertions, 0 deletions

diff --git a/scripts/postprocessing_upscale.py b/scripts/postprocessing_upscale.py
index eb42a29e..ed709688 100644
--- a/scripts/postprocessing_upscale.py
+++ b/scripts/postprocessing_upscale.py
@@ -81,6 +81,14 @@ class ScriptPostprocessingUpscale(scripts_postprocessing.ScriptPostprocessing):
 

         return image

 

+    def process_firstpass(self, pp: scripts_postprocessing.PostprocessedImage, upscale_mode=1, upscale_by=2.0, upscale_to_width=None, upscale_to_height=None, upscale_crop=False, upscaler_1_name=None, upscaler_2_name=None, upscaler_2_visibility=0.0):

+        if upscale_mode == 1:

+            pp.shared.target_width = upscale_to_width

+            pp.shared.target_height = upscale_to_height

+        else:

+            pp.shared.target_width = int(pp.image.width * upscale_by)

+            pp.shared.target_height = int(pp.image.height * upscale_by)

+

     def process(self, pp: scripts_postprocessing.PostprocessedImage, upscale_mode=1, upscale_by=2.0, upscale_to_width=None, upscale_to_height=None, upscale_crop=False, upscaler_1_name=None, upscaler_2_name=None, upscaler_2_visibility=0.0):

         if upscaler_1_name == "None":

             upscaler_1_name = None

@@ -126,6 +134,10 @@ class ScriptPostprocessingUpscaleSimple(ScriptPostprocessingUpscale):
             "upscaler_name": upscaler_name,

         }

 

+    def process_firstpass(self, pp: scripts_postprocessing.PostprocessedImage, upscale_by=2.0, upscaler_name=None):

+        pp.shared.target_width = int(pp.image.width * upscale_by)

+        pp.shared.target_height = int(pp.image.height * upscale_by)

+

     def process(self, pp: scripts_postprocessing.PostprocessedImage, upscale_by=2.0, upscaler_name=None):

         if upscaler_name is None or upscaler_name == "None":

             return