From 42f86c8c09f631153ded77e94c626ae7b22a32c0 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Thu, 8 Dec 2022 06:00:48 -0800
Subject: [PATCH] Update davit.py

---
 timm/models/davit.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/timm/models/davit.py b/timm/models/davit.py
index 16cc99f4..09946fcf 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -111,7 +111,7 @@ class PatchEmbed(nn.Module):
             self.norm = nn.LayerNorm(in_chans)
 
 
-    def forward(self, x, size: Tuple[int, int]):
+    def forward(self, x : Tensor, size: Tuple[int, int]):
         H, W = size
         dim = len(x.shape)
         if dim == 3: