Revert "fea(): fixed another pytest for gptj"

This reverts commit 21fdeae.
imangohari1 · Aug 8, 2024 · fc9ddcf · fc9ddcf
1 parent 21fdeae
commit fc9ddcf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optimum/habana/transformers/models/gptj/modeling_gptj.py b/optimum/habana/transformers/models/gptj/modeling_gptj.py
@@ -160,12 +160,12 @@ def forward(
         position_ids: Optional[torch.LongTensor] = None,
         head_mask: Optional[torch.FloatTensor] = None,
         use_cache: Optional[bool] = False,
-        sin: Optional[torch.Tensor] = None,
-        cos: Optional[torch.Tensor] = None,
         output_attentions: Optional[bool] = False,
         token_idx: Optional[torch.Tensor] = None,
         reuse_cache: Optional[bool] = False,
         cache_idx: Optional[int] = None,
+        sin: Optional[torch.Tensor] = None,
+        cos: Optional[torch.Tensor] = None,
     ) -> Union[
         Tuple[torch.Tensor, Tuple[torch.Tensor]],
         Optional[Tuple[torch.Tensor, Tuple[torch.Tensor], Tuple[torch.Tensor, ...]]],