Unverified Commit 623b4f7c authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

[VisionTextDualEncoder] Add token_type_ids param (#15073)



* fix doc example - TypeError: get_text_features() got an unexpected keyword argument 'token_type_ids'

* add token_type_ids param
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent ac224bb0
...@@ -214,6 +214,7 @@ class VisionTextDualEncoderModel(PreTrainedModel): ...@@ -214,6 +214,7 @@ class VisionTextDualEncoderModel(PreTrainedModel):
input_ids=None, input_ids=None,
attention_mask=None, attention_mask=None,
position_ids=None, position_ids=None,
token_type_ids=None,
output_attentions=None, output_attentions=None,
output_hidden_states=None, output_hidden_states=None,
return_dict=None, return_dict=None,
...@@ -238,6 +239,7 @@ class VisionTextDualEncoderModel(PreTrainedModel): ...@@ -238,6 +239,7 @@ class VisionTextDualEncoderModel(PreTrainedModel):
input_ids=input_ids, input_ids=input_ids,
attention_mask=attention_mask, attention_mask=attention_mask,
position_ids=position_ids, position_ids=position_ids,
token_type_ids=token_type_ids,
output_attentions=output_attentions, output_attentions=output_attentions,
output_hidden_states=output_hidden_states, output_hidden_states=output_hidden_states,
return_dict=return_dict, return_dict=return_dict,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment