Browse Source

fix: add support for additional keyword arguments in modeling_unimer_mbart.py to adaptation transformers 4.54.1

myhloli 3 tháng trước cách đây
mục cha
commit
f7affcefb7

+ 1 - 0
mineru/model/mfr/unimernet/unimernet_hf/unimer_mbart/modeling_unimer_mbart.py

@@ -2191,6 +2191,7 @@ class UnimerMBartForCausalLM(UnimerMBartPreTrainedModel, GenerationMixin):
         output_hidden_states: Optional[bool] = None,
         return_dict: Optional[bool] = None,
         count_gt: Optional[torch.LongTensor] = None,
+        **kwargs,
     ) -> Union[Tuple, CausalLMOutputWithCrossAttentions]:
         r"""
         Args: