Update modeling_alinlight.py

modeling_alinlight.py

@@ -285,7 +285,8 @@ class AlinlightDecoderLayer(nn.Module):
         past_key_value=None, 
         output_attentions=False, 
         use_cache=False, 
-        rotary_pos_emb=None
+        rotary_pos_emb=None,
+        **kwargs,
     ):
         residual = hidden_states
         hidden_states = self.input_layernorm(hidden_states)
@@ -384,6 +385,7 @@ class AlinlightModel(AlinlightPreTrainedModel):
         output_attentions: Optional[bool] = None,
         output_hidden_states: Optional[bool] = None,
         return_dict: Optional[bool] = None,
+        **kwargs,
     ):
         output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
         output_hidden_states = output_hidden_states if output_hidden_states is not None else self.config.output_hidden_states