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

Fix incorrect type annotation in get_auxiliary_logits (#37955)

Browse Source 
Correct type annotation from Dict(str, Tensor) to Dict[str, Tensor]
This commit is contained in:
Tanuj Rai 2025-05-05 23:30:49 +05:30 committed by GitHub
parent d80f53fa50
commit 46c0e1ff80
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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/mask2former/modeling_mask2former.py
View File

@ -2359,7 +2359,7 @@ class Mask2FormerForUniversalSegmentation(Mask2FormerPreTrainedModel):
return sum(loss_dict.values())
def get_auxiliary_logits(self, classes: torch.Tensor, output_masks: torch.Tensor):
auxiliary_logits: List[Dict(str, Tensor)] = []
auxiliary_logits: List[Dict[str, Tensor]] = []
for aux_binary_masks, aux_classes in zip(output_masks[:-1], classes[:-1]):
auxiliary_logits.append({"masks_queries_logits": aux_binary_masks, "class_queries_logits": aux_classes})