Преглед на файлове

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

myhloli преди 3 месеца
родител
ревизия
f7affcefb7
променени са 1 файла, в които са добавени 1 реда и са изтрити 0 реда
  1. 1 0
      mineru/model/mfr/unimernet/unimernet_hf/unimer_mbart/modeling_unimer_mbart.py

+ 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: