Update modeling_challenger.py

modeling_challenger.py

@@ -108,7 +108,16 @@ class ChallengerForCausalLM(PreTrainedModel, GenerationMixin):
         super().__init__(config)
         self.model = GPT(config)
 
-    def forward(self, input_ids, labels=None):
+    def forward(
+        self,
+        input_ids=None,
+        inputs_embeds=None,
+        attention_mask=None,
+        labels=None,
+        use_cache=None,
+        past_key_values=None,
+        **kwargs
+    ):
         logits, loss = self.model(input_ids, labels)
         return CausalLMOutputWithCrossAttentions(
             loss=loss,