"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "d7d60df0ec1f9aa1ee49018736fd7da19160e820"
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""" ...@@ -942,6 +942,12 @@ SWIN_INPUTS_DOCSTRING = r"""
@add_start_docstrings( @add_start_docstrings(
"The bare Swin Model transformer outputting raw hidden-states without any specific head on top.", "The bare Swin Model transformer outputting raw hidden-states without any specific head on top.",
SWIN_START_DOCSTRING, 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): class SwinModel(SwinPreTrainedModel):
def __init__(self, config, add_pooling_layer=True, use_mask_token=False): def __init__(self, config, add_pooling_layer=True, use_mask_token=False):
......
...@@ -499,7 +499,6 @@ OBJECTS_TO_IGNORE = [ ...@@ -499,7 +499,6 @@ OBJECTS_TO_IGNORE = [
"SqueezeBertTokenizerFast", "SqueezeBertTokenizerFast",
"SummarizationPipeline", "SummarizationPipeline",
"Swin2SRImageProcessor", "Swin2SRImageProcessor",
"SwinModel",
"Swinv2Model", "Swinv2Model",
"SwitchTransformersConfig", "SwitchTransformersConfig",
"T5Config", "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