Unverified Commit cc44ca80 authored by Shivanand's avatar Shivanand Committed by GitHub
Browse files

[docstring] `SwinModel` docstring fix (#26679)



* remove from utils

* updated doc string

* only in the model

* Update src/transformers/models/swin/modeling_swin.py
Co-authored-by: default avatarYih-Dar <2521628+ydshieh@users.noreply.github.com>

* Update src/transformers/models/swin/modeling_swin.py
Co-authored-by: default avatarYih-Dar <2521628+ydshieh@users.noreply.github.com>

---------
Co-authored-by: default avatarYih-Dar <2521628+ydshieh@users.noreply.github.com>
parent da69de17
......@@ -942,6 +942,12 @@ SWIN_INPUTS_DOCSTRING = r"""
@add_start_docstrings(
"The bare Swin Model transformer outputting raw hidden-states without any specific head on top.",
SWIN_START_DOCSTRING,
"""
add_pooling_layer (`bool`, *optional*, defaults to `True`):
Whether or not to apply pooling layer.
use_mask_token (`bool`, *optional*, defaults to `False`):
Whether or not to create and apply mask tokens in the embedding layer.
""",
)
class SwinModel(SwinPreTrainedModel):
def __init__(self, config, add_pooling_layer=True, use_mask_token=False):
......
......@@ -499,7 +499,6 @@ OBJECTS_TO_IGNORE = [
"SqueezeBertTokenizerFast",
"SummarizationPipeline",
"Swin2SRImageProcessor",
"SwinModel",
"Swinv2Model",
"SwitchTransformersConfig",
"T5Config",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment