瀏覽代碼

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: