fix(api): use tile size for all highres scaling methods

ssube · Jun 15, 2023 · 7b0bed7 · 7b0bed7
1 parent dcea248
commit 7b0bed7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/api/onnx_web/diffusers/run.py b/api/onnx_web/diffusers/run.py
@@ -147,15 +147,17 @@ def run_highres(
     )
 
     def highres_tile(tile: Image.Image, dims):
+        scaled_size = (size.height * highres.scale, size.width * highres.scale)
+
         if highres.method == "bilinear":
             logger.debug("using bilinear interpolation for highres")
             tile = tile.resize(
-                (size.height, size.width), resample=Image.Resampling.BILINEAR
+                scaled_size, resample=Image.Resampling.BILINEAR
             )
         elif highres.method == "lanczos":
             logger.debug("using Lanczos interpolation for highres")
             tile = tile.resize(
-                (size.height, size.width), resample=Image.Resampling.LANCZOS
+                scaled_size, resample=Image.Resampling.LANCZOS
             )
         else:
             logger.debug("using upscaling pipeline for highres")