fix compatibility with deep shrink

2023-11-20 20:23:57 +01:00
parent ee85b4cff6
commit 2255ae7660
1 changed files with 1 additions and 1 deletions
@@ -278,7 +278,7 @@ class CrossAttentionPatch:

                if mask is not None:
                    # TODO: needs testing
-                    for rate in [1, 2, 4, 8]:
+                    for rate in [1, 2, 4, 8, 16]:
                        mask_h = -(-lh//rate) # fancy ceil
                        mask_w = -(-lw//rate)