shmuli commited on
Commit
441184b
·
verified ·
1 Parent(s): 420ba7d

Trying to fix issues with extra arguments to the model

Browse files
Files changed (1) hide show
  1. modeling_decilm.py +1 -0
modeling_decilm.py CHANGED
@@ -503,6 +503,7 @@ class DeciLMFlashAttention2(DeciLMAttention):
503
  use_cache: bool = False,
504
  cache_position: Optional[torch.LongTensor] = None,
505
  position_embeddings: Optional[Tuple[torch.Tensor, torch.Tensor]] = None, # will become mandatory in v4.45
 
506
  ) -> Tuple[torch.Tensor, Optional[torch.Tensor], Optional[Tuple[torch.Tensor]]]:
507
  output_attentions = False
508
 
 
503
  use_cache: bool = False,
504
  cache_position: Optional[torch.LongTensor] = None,
505
  position_embeddings: Optional[Tuple[torch.Tensor, torch.Tensor]] = None, # will become mandatory in v4.45
506
+ **kwargs,
507
  ) -> Tuple[torch.Tensor, Optional[torch.Tensor], Optional[Tuple[torch.Tensor]]]:
508
  output_attentions = False
509