prince-canuma commited on
Commit
9c54974
·
verified ·
1 Parent(s): 4e15566

Fix create mask typeError

Browse files
Files changed (1) hide show
  1. 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