Update davit.py

timm/models/davit.py

@@ -36,7 +36,7 @@ from .registry import register_model
 __all__ = ['DaViT']
 
 class MySequential(nn.Sequential):
-    def forward(self, inputs : Tuple[Tensor, Tensor]):
+    def forward(self, inputs : Tuple[Tensor, Tuple[int, int]]):
         for module in self._modules.values():
             #if type(inputs) == tuple:
             inputs = module(*inputs)