Skip to content

GitLab

Unverified Commit ef74da65 authored by David Reguera's avatar David Reguera Committed by GitHub
Browse files

16059 - Add extra type hints for AltCLIPModel (#25399)

parent f456b4d1
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/transformers/models/altclip/modeling_altclip.py
View file @ ef74da65
...@@ -1420,7 +1420,7 @@ class AltCLIPTextModel(AltCLIPPreTrainedModel): ...@@ -1420,7 +1420,7 @@ class AltCLIPTextModel(AltCLIPPreTrainedModel):
output_attentions: Optional[bool] = None, output_attentions: Optional[bool] = None,
return_dict: Optional[bool] = None, return_dict: Optional[bool] = None,
output_hidden_states: Optional[bool] = None, output_hidden_states: Optional[bool] = None,
): ) -> Union[Tuple, BaseModelOutputWithPoolingAndProjection]:
r""" r"""
Returns: Returns:
...@@ -1613,7 +1613,7 @@ class AltCLIPModel(AltCLIPPreTrainedModel): ...@@ -1613,7 +1613,7 @@ class AltCLIPModel(AltCLIPPreTrainedModel):
pixel_values: Optional[torch.FloatTensor] = None, pixel_values: Optional[torch.FloatTensor] = None,
attention_mask: Optional[torch.Tensor] = None, attention_mask: Optional[torch.Tensor] = None,
position_ids: Optional[torch.LongTensor] = None, position_ids: Optional[torch.LongTensor] = None,
token_type_ids=None, token_type_ids: Optional[torch.Tensor] = None,
return_loss: Optional[bool] = None, return_loss: Optional[bool] = None,
output_attentions: Optional[bool] = None, output_attentions: Optional[bool] = None,
output_hidden_states: Optional[bool] = None, output_hidden_states: Optional[bool] = None,
......
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