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

fixes

Browse Source
This commit is contained in:
Arthur 2025-07-03 15:36:52 +02:00
parent 5af5bccd56
commit 9968c85e4b
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/models/sam_hq/modeling_sam_hq.py
View File

@ -1517,9 +1517,9 @@ class SamHQModel(SamHQPreTrainedModel):
return SamHQImageSegmentationOutput(
iou_scores=mask_decoder_output.iou_scores,
pred_masks=mask_decoder_output.masks,
mask_decoder_attention=mask_decoder_output.mask_decoder_attentions,
mask_decoder_attentions=mask_decoder_output.mask_decoder_attentions,
vision_hidden_states=vision_outputs.hidden_states,
vision_attentions=vision_outputs.vision_attentions,
vision_attentions=vision_outputs.attentions,
)

4
src/transformers/models/sam_hq/modular_sam_hq.py
View File

@ -610,9 +610,9 @@ class SamHQModel(SamModel):
return SamHQImageSegmentationOutput(
iou_scores=mask_decoder_output.iou_scores,
pred_masks=mask_decoder_output.masks,
mask_decoder_attention=mask_decoder_output.mask_decoder_attentions,
mask_decoder_attentions=mask_decoder_output.mask_decoder_attentions,
vision_hidden_states=vision_outputs.hidden_states,
vision_attentions=vision_outputs.vision_attentions,
vision_attentions=vision_outputs.attentions,
)