Skip to content

Commit 0834563

Browse files
committed
fix
1 parent 50eb37a commit 0834563

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

src/mcore_bridge/model/mm_gpts/gemma4.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,7 @@ def get_inputs_embeds(self, inputs_embeds, **kwargs):
132132
audio_mask_e = audio_mask.unsqueeze(-1).expand_as(inputs_embeds).to(inputs_embeds.device)
133133
inputs_embeds = inputs_embeds.masked_scatter(audio_mask_e, audio_features)
134134
res['inputs_embeds'] = inputs_embeds
135+
res['llm_input_ids'] = input_ids
135136
return res
136137

137138

0 commit comments

Comments
 (0)