Update app.py

app.py

@@ -117,9 +117,22 @@ def inpaint(
         def callback_on_step_end(pipe_self, i, t, callback_kwargs):
             latents = callback_kwargs.get("latents", None)
             if latents is not None:
-                latents = latents * mask_t + latents_orig * (1 - mask_t)
+                # Ajustar dinámicamente los tamaños al del tensor actual
+                if mask_t.shape[-2:] != latents.shape[-2:]:
+                    resized_mask = torch.nn.functional.interpolate(mask_t, size=latents.shape[-2:], mode="nearest")
+                else:
+                    resized_mask = mask_t
+        
+                if latents_orig.shape[-2:] != latents.shape[-2:]:
+                    resized_latents_orig = torch.nn.functional.interpolate(latents_orig, size=latents.shape[-2:], mode="nearest")
+                else:
+                    resized_latents_orig = latents_orig
+        
+                latents = latents * resized_mask + resized_latents_orig * (1 - resized_mask)
                 callback_kwargs["latents"] = latents
-            print("callback_kwargsssssssssssssss", callback_kwargs.keys())
+        
+            # 🔍 Solo para depuración
+            print(f"[Callback] step={i}, keys={list(callback_kwargs.keys())}, latents={latents.shape if latents is not None else None}")
             return callback_kwargs
     
         callback_on_step_end_tensor_inputs = ["latents"]