From eb94efb21817eb68629cfab63083bfa34ddddd70 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Wojtek=20Jasi=C5=84ski?= <wojciech.jasinski@swmansion.com>
Date: Fri, 1 Nov 2024 23:27:06 +0100
Subject: [PATCH] fix pos embed dynamic resampling for eva

---
 timm/models/eva.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/timm/models/eva.py b/timm/models/eva.py
index 62e986ba..fe871540 100644
--- a/timm/models/eva.py
+++ b/timm/models/eva.py
@@ -560,9 +560,11 @@ class Eva(nn.Module):
         if self.dynamic_img_size:
             B, H, W, C = x.shape
             if self.pos_embed is not None:
+                prev_grid_size = self.patch_embed.grid_size
                 pos_embed = resample_abs_pos_embed(
                     self.pos_embed,
-                    (H, W),
+                    new_size=(H, W),
+                    old_size=prev_grid_size,
                     num_prefix_tokens=self.num_prefix_tokens,
                 )
             else: