riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

Conditional DETR type hint fix (#25505)

Browse Source
This commit is contained in:
Matt 2023-08-14 18:12:06 +01:00 committed by GitHub
parent c41291965f
commit b42010bb1d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/conditional_detr/modeling_conditional_detr.py
View File

@ -1534,7 +1534,7 @@ class ConditionalDetrModel(ConditionalDetrPreTrainedModel):
output_attentions: Optional[bool] = None,
output_hidden_states: Optional[bool] = None,
return_dict: Optional[bool] = None,
) -> Union[torch.FloatTensor, ConditionalDetrModelOutput]:
) -> Union[Tuple[torch.FloatTensor], ConditionalDetrModelOutput]:
r"""
Returns: