From 28b712cf5bdb8e53f231b27ef91ab3961f5fddc1 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Thu, 8 Dec 2022 10:37:52 -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 273ebcd0..58931cfd 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -341,7 +341,7 @@ class SpatialBlock(nn.Module):
         x = self.cpe[1](x, size)
         if self.ffn:
             x = x + self.drop_path(self.mlp(self.norm2(x)))
-        return x : Tensor, size
+        return x, size