diff --git a/src/mcore_bridge/patcher.py b/src/mcore_bridge/patcher.py index 392ff34..c43141d 100644 --- a/src/mcore_bridge/patcher.py +++ b/src/mcore_bridge/patcher.py @@ -423,6 +423,7 @@ def forward( input_ids=input_ids, position_ids=position_ids, embedding=embedding, + packed_seq_params=packed_seq_params, hidden_states=hidden_states, ) assert not self.transformer_layer.self_attention.config.apply_rope_fusion