Update app.py

app.py

@@ -110,6 +110,11 @@ def tiled_flux_img2img(pipe, prompt, image, strength, steps, guidance, generator
                 continue
             tile = image.crop((x, y, x + tile_w, y + tile_h))
 
+            # Force tile to div by 16
+            new_tile_w = make_divisible_by_16(tile_w)
+            new_tile_h = make_divisible_by_16(tile_h)
+            tile = tile.resize((new_tile_w, new_tile_h), resample=Image.LANCZOS)
+
             # Run Flux on tile
             gen_tile = pipe(
                 prompt=prompt,
@@ -117,14 +122,13 @@ def tiled_flux_img2img(pipe, prompt, image, strength, steps, guidance, generator
                 strength=strength,
                 num_inference_steps=steps,
                 guidance_scale=guidance,
-                height=tile_h,
-                width=tile_w,
+                height=new_tile_h,
+                width=new_tile_w,
                 generator=generator,
             ).images[0]
 
-            # Resize gen_tile back to original tile dimensions if pipeline resized it
-            if gen_tile.size != (tile_w, tile_h):
-                gen_tile = gen_tile.resize((tile_w, tile_h), resample=Image.LANCZOS)
+            # Resize gen_tile back to original tile dimensions
+            gen_tile = gen_tile.resize((tile_w, tile_h), resample=Image.LANCZOS)
 
             # Paste with blending if overlap
             if overlap > 0: