From 7ef489adc91efd8f7f3580ea0b1e03d421f967df Mon Sep 17 00:00:00 2001
From: GeorgLegato <bulbul@onlinehome.de>
Date: Sat, 15 Apr 2023 20:38:04 +0200
Subject: [PATCH] fix width for square

---
 scripts/inifnite-zoom.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/scripts/inifnite-zoom.py b/scripts/inifnite-zoom.py
index 9353ec0..c6d62f8 100644
--- a/scripts/inifnite-zoom.py
+++ b/scripts/inifnite-zoom.py
@@ -7,6 +7,7 @@ sys.path.extend(basedir + "/extensions/infinite-zoom-automatic1111-webui/")
 import numpy as np
 import gradio as gr
 from PIL import Image
+import math
 
 from iz_helpers import shrink_and_paste_on_blank, write_video
 from webui import wrap_gradio_gpu_call
@@ -140,7 +141,8 @@ def create_zoom(
             height,
         )
         current_image = processed.images[0]
-    mask_width = 128
+
+    mask_width = math.trunc(width/4)  # was initially 512px => 128px
     num_interpol_frames = round(video_frame_rate * zoom_speed)
 
     all_frames = []