Skip to content

Commit 356b3cd

Browse files
authored
Fix missing return type for MLCD docs (#37527)
* Fix missing return type for docs * trigger
1 parent 0ad3710 commit 356b3cd

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

src/transformers/models/mlcd/modeling_mlcd.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
add_start_docstrings_to_model_forward,
3434
can_return_tuple,
3535
logging,
36+
replace_return_docstrings,
3637
torch_int,
3738
)
3839
from .configuration_mlcd import MLCDVisionConfig
@@ -631,6 +632,7 @@ def get_input_embeddings(self) -> nn.Module:
631632
return self.vision_model.embeddings.patch_embedding
632633

633634
@add_start_docstrings_to_model_forward(MLCD_VISION_INPUTS_DOCSTRING)
635+
@replace_return_docstrings(output_type=BaseModelOutputWithPooling, config_class=MLCDVisionConfig)
634636
def forward(
635637
self,
636638
pixel_values: Optional[torch.FloatTensor] = None,

src/transformers/models/mlcd/modular_mlcd.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
add_start_docstrings,
3333
add_start_docstrings_to_model_forward,
3434
logging,
35+
replace_return_docstrings,
3536
)
3637
from ..clip.modeling_clip import (
3738
CLIPMLP,
@@ -548,6 +549,7 @@ def _init_weights(self, module):
548549
)
549550
class MLCDVisionModel(CLIPVisionModel):
550551
@add_start_docstrings_to_model_forward(MLCD_VISION_INPUTS_DOCSTRING)
552+
@replace_return_docstrings(output_type=BaseModelOutputWithPooling, config_class=MLCDVisionConfig)
551553
def forward(
552554
self,
553555
pixel_values: Optional[torch.FloatTensor] = None,

0 commit comments

Comments
 (0)