Update app.py

app.py

@@ -146,12 +146,17 @@ def infer_60(
     pyx.upload_to_ftp(sd35_path)
     upscaler_2.to(torch.device('cuda'))
     with torch.no_grad():
-        upscale2 = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
+        upscale = upscaler_2(sd_image, tiling=True, tile_width=512, tile_height=512)
+        upscale2 = upscaler_2(upscale, tiling=True, tile_width=512, tile_height=512)
     print('-- got upscaled image --')
-    downscale2 = upscale2.resize((upscale2.width // 4, upscale2.height // 4),Image.LANCZOS)
+    downscale = upscale2.resize((upscale2.width // 4, upscale2.height // 4),Image.LANCZOS)
+    downscale2 = downscale.resize((downscale.width // 4, downscale.height // 4),Image.LANCZOS)
     upscale_path = f"sd35ll_upscale_{timestamp}.png"
-    downscale2.save(upscale_path,optimize=False,compress_level=0)
+    upscale2_path = f"sd35ll_upscale2_{timestamp}.png"
+    downscale.save(upscale_path,optimize=False,compress_level=0)
+    downscale2.save(upscale2_path,optimize=False,compress_level=0)
     pyx.upload_to_ftp(upscale_path)
+    pyx.upload_to_ftp(upscale2_path)
     return sd_image, prompt
 
 @spaces.GPU(duration=100)
@@ -196,12 +201,17 @@ def infer_90(
     pyx.upload_to_ftp(sd35_path)
     upscaler_2.to(torch.device('cuda'))
     with torch.no_grad():
-        upscale2 = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
+        upscale = upscaler_2(sd_image, tiling=True, tile_width=512, tile_height=512)
+        upscale2 = upscaler_2(upscale, tiling=True, tile_width=512, tile_height=512)
     print('-- got upscaled image --')
-    downscale2 = upscale2.resize((upscale2.width // 4, upscale2.height // 4),Image.LANCZOS)
+    downscale = upscale2.resize((upscale2.width // 4, upscale2.height // 4),Image.LANCZOS)
+    downscale2 = downscale.resize((downscale.width // 4, downscale.height // 4),Image.LANCZOS)
     upscale_path = f"sd35ll_upscale_{timestamp}.png"
-    downscale2.save(upscale_path,optimize=False,compress_level=0)
+    upscale2_path = f"sd35ll_upscale2_{timestamp}.png"
+    downscale.save(upscale_path,optimize=False,compress_level=0)
+    downscale2.save(upscale2_path,optimize=False,compress_level=0)
     pyx.upload_to_ftp(upscale_path)
+    pyx.upload_to_ftp(upscale2_path)
     return sd_image, prompt
 
 @spaces.GPU(duration=120)
@@ -246,12 +256,17 @@ def infer_110(
     pyx.upload_to_ftp(sd35_path)
     upscaler_2.to(torch.device('cuda'))
     with torch.no_grad():
-        upscale2 = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
+        upscale = upscaler_2(sd_image, tiling=True, tile_width=512, tile_height=512)
+        upscale2 = upscaler_2(upscale, tiling=True, tile_width=512, tile_height=512)
     print('-- got upscaled image --')
-    downscale2 = upscale2.resize((upscale2.width // 4, upscale2.height // 4),Image.LANCZOS)
+    downscale = upscale2.resize((upscale2.width // 4, upscale2.height // 4),Image.LANCZOS)
+    downscale2 = downscale.resize((downscale.width // 4, downscale.height // 4),Image.LANCZOS)
     upscale_path = f"sd35ll_upscale_{timestamp}.png"
-    downscale2.save(upscale_path,optimize=False,compress_level=0)
+    upscale2_path = f"sd35ll_upscale2_{timestamp}.png"
+    downscale.save(upscale_path,optimize=False,compress_level=0)
+    downscale2.save(upscale2_path,optimize=False,compress_level=0)
     pyx.upload_to_ftp(upscale_path)
+    pyx.upload_to_ftp(upscale2_path)
     return sd_image, prompt
     
 css = """