fix(api): use correct tile size for highres upscaling

ssube · Jun 17, 2023 · 12e489b · 12e489b
1 parent 5713957
commit 12e489b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/onnx_web/diffusers/run.py b/api/onnx_web/diffusers/run.py
@@ -147,7 +147,7 @@ def run_highres(
     )
 
     def highres_tile(tile: Image.Image, dims):
-        scaled_size = (size.width * highres.scale, size.height * highres.scale)
+        scaled_size = (tile.width * highres.scale, tile.height * highres.scale)
 
         if highres.method == "bilinear":
             logger.debug("using bilinear interpolation for highres")