Update app.py

app.py

@@ -13,11 +13,11 @@ vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=to
 pipe = LucyEditPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
 pipe.to("cuda")
 
-def calculate_resolution(input_width, input_height, min_dimension=480, max_dimension=832):
+def calculate_resolution(input_width, input_height, min_dimension=480, max_dimension=832, compatible_round=32):
     """Calculate optimal resolution preserving aspect ratio within bounds"""
-    # Ensure dimensions are multiples of 16
-    def round_to_16(x):
-        return max(min_dimension, min(max_dimension, int(round(x / 16.0) * 16)))
+    # Ensure dimensions are multiples of the compatible rounding
+    def round_to(x, compatible_round):
+        return max(min_dimension, min(max_dimension, int(round(x / compatible_round) * compatible_round)))
     
     # Get aspect ratio
     aspect_ratio = input_width / input_height
@@ -40,7 +40,7 @@ def calculate_resolution(input_width, input_height, min_dimension=480, max_dimen
             if new_width > max_dimension:
                 new_width = max_dimension
         
-        return round_to_16(new_width), round_to_16(new_height)
+        return round_to(new_width, compatible_round), round_to(new_height, compatible_round)
     
     # Portrait videos (height > width)
     else:
@@ -56,7 +56,7 @@ def calculate_resolution(input_width, input_height, min_dimension=480, max_dimen
             if new_height > max_dimension:
                 new_height = max_dimension
         
-        return round_to_16(new_width), round_to_16(new_height)
+        return round_to(new_width, compatible_round), round_to(new_height, compatible_round)
 
 
 @spaces.GPU(duration=120)