Final change

src/transformers/models/aimv2/modeling_aimv2.py

@@ -253,6 +253,7 @@ class Aimv2Attention(nn.Module):
         self,
         hidden_states: torch.Tensor,
         attention_mask: Optional[torch.Tensor] = None,
+        **kwargs,
     ) -> tuple[torch.Tensor, Optional[torch.Tensor]]:
         """Input shape: Batch x Time x Channel"""
 

tests/models/aimv2/test_modeling_aimv2.py

@@ -677,7 +677,6 @@ class AIMv2ModelIntegrationTest(unittest.TestCase):
 @require_vision
 @require_torch
 class AIMv2VisionModelIntegrationTests(unittest.TestCase):
-
     @slow
     def test_inference(self):
         model_name = "yaswanthgali/aimv2-large-patch14-224-HF"