File tree Expand file tree Collapse file tree 2 files changed +4
-0
lines changed
src/transformers/models/mlcd Expand file tree Collapse file tree 2 files changed +4
-0
lines changed Original file line number Diff line number Diff line change 3333 add_start_docstrings_to_model_forward ,
3434 can_return_tuple ,
3535 logging ,
36+ replace_return_docstrings ,
3637 torch_int ,
3738)
3839from .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 ,
Original file line number Diff line number Diff line change 3232 add_start_docstrings ,
3333 add_start_docstrings_to_model_forward ,
3434 logging ,
35+ replace_return_docstrings ,
3536)
3637from ..clip .modeling_clip import (
3738 CLIPMLP ,
@@ -548,6 +549,7 @@ def _init_weights(self, module):
548549)
549550class 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 ,
You can’t perform that action at this time.
0 commit comments