"src/git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "d4481c24558aba643ed0a20d6d06a74dd29ac636"
Unverified Commit 036e8085 authored by IMvision12's avatar IMvision12 Committed by GitHub
Browse files

Added type hints to `DebertaV2ForMultipleChoice` Pytorch (#19536)

* Update modeling_deberta_v2.py

* Update modeling_deberta_v2.py
parent 7180e172
......@@ -1616,16 +1616,16 @@ class DebertaV2ForMultipleChoice(DebertaV2PreTrainedModel):
)
def forward(
self,
input_ids=None,
attention_mask=None,
token_type_ids=None,
position_ids=None,
inputs_embeds=None,
labels=None,
output_attentions=None,
output_hidden_states=None,
return_dict=None,
):
input_ids: Optional[torch.Tensor] = None,
attention_mask: Optional[torch.Tensor] = None,
token_type_ids: Optional[torch.Tensor] = None,
position_ids: Optional[torch.Tensor] = None,
inputs_embeds: Optional[torch.Tensor] = None,
labels: Optional[torch.Tensor] = None,
output_attentions: Optional[bool] = None,
output_hidden_states: Optional[bool] = None,
return_dict: Optional[bool] = None,
) -> Union[Tuple, MultipleChoiceModelOutput]:
r"""
labels (`torch.LongTensor` of shape `(batch_size,)`, *optional*):
Labels for computing the multiple choice classification loss. Indices should be in `[0, ...,
......
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