katuni4ka commited on
Commit
971e46d
1 Parent(s): ce00379

Update modeling_jais.py

Browse files
Files changed (1) hide show
  1. modeling_jais.py +0 -1
modeling_jais.py CHANGED
@@ -1116,7 +1116,6 @@ class JAISLMHeadModel(JAISPreTrainedModel):
1116
  input_ids,
1117
  past_key_values=past_key_values,
1118
  attention_mask=attention_mask,
1119
- token_type_ids=token_type_ids,
1120
  position_ids=position_ids,
1121
  head_mask=head_mask,
1122
  inputs_embeds=inputs_embeds,
 
1116
  input_ids,
1117
  past_key_values=past_key_values,
1118
  attention_mask=attention_mask,
 
1119
  position_ids=position_ids,
1120
  head_mask=head_mask,
1121
  inputs_embeds=inputs_embeds,