Fix polarity on prefix remove

rwightman · rwightman · commit 4d9c87bbb2b8 · 2025-06-05T12:17:59.000-07:00
diff --git a/timm/models/naflexvit.py b/timm/models/naflexvit.py
@@ -1199,7 +1199,7 @@ def _pool(
                 q_len=1,
                 dtype=x.dtype,
             )
-            if self.pool_include_prefix:
+            if not self.pool_include_prefix:
                 x = x[:, self.num_prefix_tokens:]
             x = self.attn_pool(x, attn_mask=attn_mask)
             return x

Original file line number	Diff line number	Diff line change
`@@ -1199,7 +1199,7 @@ def _pool(`
`1199`	`1199`	`q_len=1,`
`1200`	`1200`	`dtype=x.dtype,`
`1201`	`1201`	`)`
`1202`		`- if self.pool_include_prefix:`
	`1202`	`+ if not self.pool_include_prefix:`
`1203`	`1203`	`x = x[:, self.num_prefix_tokens:]`
`1204`	`1204`	`x = self.attn_pool(x, attn_mask=attn_mask)`
`1205`	`1205`	`return x`