"docs/git@developer.sourcefind.cn:Wenxuan/LightX2V.git" did not exist on "1e4226634c253d84ead3017da663dedd3847309f"
Unverified Commit 892399c5 authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Update AlbertModel type annotation (#24450)

Update type annotation
parent be2d9f2e
...@@ -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