Update davit.py

timm/models/davit.py

@@ -38,9 +38,9 @@ class MySequential(nn.Sequential):
 
     def forward(self, inputs : Tensor, size : Tuple[int, int]):
         for module in self:
-            inputs : Tensor
+            output = module(inputs, size)
-            size : Tuple[int, int]
+            inputs : Tensor = output[0]
-            (inputs, size) = module(inputs, size)
+            size : Tuple[int, int] = output[1]
         return inputs
 
 class ConvPosEnc(nn.Module):