katuni4ka
/

tiny-random-jais

Text Generation

Inference Endpoints

Model card Files Files and versions Community

katuni4ka commited on Oct 28

Commit

ce00379

•

1 Parent(s): 421cf1d

Update modeling_jais.py

Files changed (1) hide show

modeling_jais.py +1 -1

modeling_jais.py CHANGED Viewed

@@ -1383,7 +1383,7 @@ class JAISForTokenClassification(JAISPreTrainedModel):
             input_ids,
             past_key_values=past_key_values,
             attention_mask=attention_mask,
-            token_type_ids=token_type_ids,
             position_ids=position_ids,
             head_mask=head_mask,
             inputs_embeds=inputs_embeds,

             input_ids,
             past_key_values=past_key_values,
             attention_mask=attention_mask,
+            token_type_ids=None,
             position_ids=position_ids,
             head_mask=head_mask,
             inputs_embeds=inputs_embeds,