Update modeling_jais.py
Browse files- 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,
|