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

Fix missing return type for MLCD docs (#37527)

Browse Source 
* Fix missing return type for docs

* trigger
This commit is contained in:
Pavel Iakubovskii 2025-04-15 14:04:16 +01:00 committed by GitHub
parent 0ad3710d47
commit 356b3cd71d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/mlcd/modeling_mlcd.py
View File

@ -33,6 +33,7 @@ from ...utils import (
add_start_docstrings_to_model_forward, add_start_docstrings_to_model_forward,
can_return_tuple, can_return_tuple,
logging, logging,
replace_return_docstrings,
torch_int, torch_int,
) )
from .configuration_mlcd import MLCDVisionConfig from .configuration_mlcd import MLCDVisionConfig
@ -631,6 +632,7 @@ class MLCDVisionModel(MLCDPreTrainedModel):
return self.vision_model.embeddings.patch_embedding return self.vision_model.embeddings.patch_embedding
@add_start_docstrings_to_model_forward(MLCD_VISION_INPUTS_DOCSTRING) @add_start_docstrings_to_model_forward(MLCD_VISION_INPUTS_DOCSTRING)
@replace_return_docstrings(output_type=BaseModelOutputWithPooling, config_class=MLCDVisionConfig)
def forward( def forward(
self, self,
pixel_values: Optional[torch.FloatTensor] = None, pixel_values: Optional[torch.FloatTensor] = None,

2
src/transformers/models/mlcd/modular_mlcd.py
View File

@ -32,6 +32,7 @@ from ...utils import (
add_start_docstrings, add_start_docstrings,
add_start_docstrings_to_model_forward, add_start_docstrings_to_model_forward,
logging, logging,
replace_return_docstrings,
) )
from ..clip.modeling_clip import ( from ..clip.modeling_clip import (
CLIPMLP, CLIPMLP,
@ -548,6 +549,7 @@ class MLCDPreTrainedModel(PreTrainedModel):
) )
class MLCDVisionModel(CLIPVisionModel): class MLCDVisionModel(CLIPVisionModel):
@add_start_docstrings_to_model_forward(MLCD_VISION_INPUTS_DOCSTRING) @add_start_docstrings_to_model_forward(MLCD_VISION_INPUTS_DOCSTRING)
@replace_return_docstrings(output_type=BaseModelOutputWithPooling, config_class=MLCDVisionConfig)
def forward( def forward(
self, self,
pixel_values: Optional[torch.FloatTensor] = None, pixel_values: Optional[torch.FloatTensor] = None,