Skip to content

Commit a3e2d54

Browse files
authored
Fix typo in docstring in modeling_sam3_tracker.py (#42438)
* Update modeling_sam3_tracker.py * Apply docstring to modular_sam3_tracker.py for consistency Added custom introduction to auto_docstring for Sam3TrackerPreTrainedModel. to match typo fix in modeling_sam3_tracker.py * Fix SAM3 tracker docstrings in modular and regenerate modeling
1 parent bf8b9e7 commit a3e2d54

File tree

2 files changed

+12
-2
lines changed

2 files changed

+12
-2
lines changed

src/transformers/models/sam3_tracker/modeling_sam3_tracker.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,12 @@ def forward(self, hidden_states):
107107
return hidden_states
108108

109109

110-
@auto_docstring
110+
@auto_docstring(
111+
custom_intro="""
112+
Segment Anything Model 3 (SAM 3) for generating segmentation masks, given an input image and
113+
input points and labels, boxes, or masks.
114+
"""
115+
)
111116
class Sam3TrackerPreTrainedModel(PreTrainedModel):
112117
config_class = Sam3TrackerConfig
113118
base_model_prefix = "sam3_tracker"

src/transformers/models/sam3_tracker/modular_sam3_tracker.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,12 @@ class Sam3TrackerFeedForward(Sam2FeedForward):
136136
pass
137137

138138

139-
@auto_docstring
139+
@auto_docstring(
140+
custom_intro="""
141+
Segment Anything Model 3 (SAM 3) for generating segmentation masks, given an input image and
142+
input points and labels, boxes, or masks.
143+
"""
144+
)
140145
class Sam3TrackerPreTrainedModel(Sam2PreTrainedModel):
141146
@torch.no_grad()
142147
def _init_weights(self, module):

0 commit comments

Comments
 (0)