Avoid deprecated bidirectional mask embed keyword

#1
Files changed (1) hide show
  1. llama_bidirectional_model.py +2 -2
llama_bidirectional_model.py CHANGED
@@ -118,8 +118,8 @@ class LlamaBidirectionalModel(LlamaModel):
118
 
119
  if _HAS_NATIVE_BIDIRECTIONAL_MASK:
120
  return create_bidirectional_mask(
121
- config=self.config,
122
- input_embeds=input_embeds,
123
  attention_mask=attention_mask,
124
  )
125
 
 
118
 
119
  if _HAS_NATIVE_BIDIRECTIONAL_MASK:
120
  return create_bidirectional_mask(
121
+ self.config,
122
+ input_embeds,
123
  attention_mask=attention_mask,
124
  )
125