Skip to content

Commit bad7890

Browse files
authored
add more args to Qwen2ForCausalLM (#11068)
1 parent bbb8e00 commit bad7890

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

paddlenlp/transformers/qwen2/modeling.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1601,6 +1601,7 @@ def forward(
16011601
output_hidden_states: Optional[bool] = None,
16021602
return_dict: Optional[bool] = None,
16031603
attn_mask_startend_row_indices=None,
1604+
**kwargs,
16041605
) -> Union[Tuple, CausalLMOutputWithPast]:
16051606
r"""
16061607
Args:

0 commit comments

Comments
 (0)