Skip to content

GitLab

Unverified Commit 892399c5 authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Update AlbertModel type annotation (#24450) 

Update type annotation
parent be2d9f2e
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
src/transformers/models/albert/modeling_albert.py
View file @ 892399c5
...@@ -687,9 +687,9 @@ class AlbertModel(AlbertPreTrainedModel): ...@@ -687,9 +687,9 @@ class AlbertModel(AlbertPreTrainedModel):
position_ids: Optional[torch.LongTensor] = None, position_ids: Optional[torch.LongTensor] = None,
head_mask: Optional[torch.FloatTensor] = None, head_mask: Optional[torch.FloatTensor] = None,
inputs_embeds: Optional[torch.FloatTensor] = None, inputs_embeds: Optional[torch.FloatTensor] = None,
output_attentions: Optional[None] = None, output_attentions: Optional[bool] = None,
output_hidden_states: Optional[None] = None, output_hidden_states: Optional[bool] = None,
return_dict: Optional[None] = None, return_dict: Optional[bool] = None,
) -> Union[BaseModelOutputWithPooling, Tuple]: ) -> Union[BaseModelOutputWithPooling, Tuple]:
output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
output_hidden_states = ( output_hidden_states = (
......
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