Fix create mask typeError
Browse files- modeling_lfm2.py +1 -0
modeling_lfm2.py
CHANGED
@@ -748,6 +748,7 @@ class LFM2Model(LFM2PretrainedModel):
|
|
748 |
attention_mask=attention_mask,
|
749 |
cache_position=cache_position,
|
750 |
past_key_values=past_key_values,
|
|
|
751 |
)
|
752 |
hidden_states = inputs_embeds
|
753 |
|
|
|
748 |
attention_mask=attention_mask,
|
749 |
cache_position=cache_position,
|
750 |
past_key_values=past_key_values,
|
751 |
+
position_ids=position_ids,
|
752 |
)
|
753 |
hidden_states = inputs_embeds
|
754 |
|