soon fixed

models/matany_compat_patch.py

@@ -49,17 +49,25 @@ def apply_matany_t1_squeeze_guard() -> bool:
         # Store original method
         orig_method = getattr(MatAnyone, method_name)
 
-        def method_compat(self, img, *args, **kwargs):
-            try:
-                if isinstance(img, torch.Tensor) and img.dim() == 5 and img.shape[1] == 1:
-                    log.info(f"[MatAnyCompat] Squeezing 5D {img.shape} to 4D {img.squeeze(1).shape} in {method_name}")
-                    img = img.squeeze(1)  # [B,1,C,H,W] → [B,C,H,W]
-            except Exception as e:
-                log.warning(f"[MatAnyCompat] Failed to process input shape in {method_name}: %s", e)
-            return orig_method(self, img, *args, **kwargs)
+        def create_method_compat(orig_fn, name):
+            def method_compat(self, img, *args, **kwargs):
+                try:
+                    if isinstance(img, torch.Tensor):
+                        # Handle [B, 1, C, H, W] -> [B, C, H, W] 
+                        if img.dim() == 5 and img.shape[1] == 1:
+                            log.info(f"[MatAnyCompat] Squeezing 5D {img.shape} to 4D {img.squeeze(1).shape} in {name}")
+                            img = img.squeeze(1)
+                        # Handle [1, 1, C, H, W] -> [C, H, W] (remove both batch and time dims)
+                        elif img.dim() == 5 and img.shape[0] == 1 and img.shape[1] == 1:
+                            log.info(f"[MatAnyCompat] Squeezing 5D {img.shape} to 3D {img.squeeze(0).squeeze(0).shape} in {name}")
+                            img = img.squeeze(0).squeeze(0)  # Remove batch and time dimensions
+                except Exception as e:
+                    log.warning(f"[MatAnyCompat] Failed to process input shape in {name}: %s", e)
+                return orig_fn(self, img, *args, **kwargs)
+            return method_compat
 
-        setattr(MatAnyone, method_name, method_compat)
-        setattr(MatAnyone, f"_{method_name}_patched", True)
+        setattr(MatAnyone, method_name, create_method_compat(orig_method, method_name))
+        setattr(MatAnyone, f"_{method_name}_patched", True)  # FIXED: Changed MatAny to MatAnyone
         log.info(f"[MatAnyCompat] Applied T=1 squeeze guard in MatAnyone.{method_name}")
         patched = True