Unverified Commit 5c1cda9d authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

fix example in config doc (#11696)

parent 77f4c46b
...@@ -164,7 +164,7 @@ class CLIPVisionConfig(PretrainedConfig): ...@@ -164,7 +164,7 @@ class CLIPVisionConfig(PretrainedConfig):
gradient_checkpointing (:obj:`bool`, `optional`, defaults to :obj:`False`): gradient_checkpointing (:obj:`bool`, `optional`, defaults to :obj:`False`):
If True, use gradient checkpointing to save memory at the expense of slower backward pass. If True, use gradient checkpointing to save memory at the expense of slower backward pass.
Example:: Example::
>>> from transformers import CLIPVisionModel, CLIPVisionConfig >>> from transformers import CLIPVisionModel, CLIPVisionConfig
...@@ -224,15 +224,14 @@ class CLIPConfig(PretrainedConfig): ...@@ -224,15 +224,14 @@ class CLIPConfig(PretrainedConfig):
outputs. Read the documentation from :class:`~transformers.PretrainedConfig` for more information. outputs. Read the documentation from :class:`~transformers.PretrainedConfig` for more information.
Args: Args:
projection_dim: (:obj:`int`, `optional`, defaults to 512): text_config_dict (:obj:`dict`, `optional`):
Dictionary of configuration options used to initialize :class:`~transformers.CLIPTextConfig`.
vision_config_dict (:obj:`dict`, `optional`):
Dictionary of configuration options used to initialize :class:`~transformers.CLIPVisionConfig`.
projection_dim (:obj:`int`, `optional`, defaults to 512):
Dimentionality of text and vision projection layers. Dimentionality of text and vision projection layers.
kwargs (`optional`): kwargs (`optional`):
Dictionary of keyword arguments. Notably: Dictionary of keyword arguments.
- **text_config** (:class:`~transformers.CLIPTextConfig`, `optional`) -- An instance of a configuration
object that defines the text model config.
- **vision_config** (:class:`~transformers.CLIPVisionConfig`, `optional`) -- An instance of a
configuration object that defines the vision model config.
""" """
model_type = "clip" model_type = "clip"
......
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