diff --git a/timm/models/davit.py b/timm/models/davit.py
index 582e4010..09b44e37 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -38,6 +38,8 @@ class MySequential(nn.Sequential):
 
     def forward(self, inputs : Tensor, size : Tuple[int, int]):
         for module in self:
+            inputs : Tensor
+            size : Tuple[int, int]
             (inputs, size) = module(inputs, size)
         return inputs