BoLiu Oliver Holworthy commited on
Commit
4234447
·
1 Parent(s): 9679e50

Avoid deprecated bidirectional mask embed keyword (#1)

Browse files

- Avoid deprecated bidirectional mask embed keyword (c06f1bf6db6000206a68ead161e10c404c1fa6d5)


Co-authored-by: Oliver Holworthy <nvidia-oliver-holworthy@users.noreply.huggingface.co>

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