Skip to content

Commit

Permalink
fix example in config doc (huggingface#11696)
Browse files Browse the repository at this point in the history
  • Loading branch information
patil-suraj authored and Iwontbecreative committed Jul 15, 2021
1 parent 0c2a63e commit 67d4654
Showing 1 changed file with 7 additions and 8 deletions.
15 changes: 7 additions & 8 deletions src/transformers/models/clip/configuration_clip.py
Original file line number Diff line number Diff line change
Expand Up @@ -164,7 +164,7 @@ class CLIPVisionConfig(PretrainedConfig):
gradient_checkpointing (:obj:`bool`, `optional`, defaults to :obj:`False`):
If True, use gradient checkpointing to save memory at the expense of slower backward pass.
Example::
Example::
>>> from transformers import CLIPVisionModel, CLIPVisionConfig
Expand Down Expand Up @@ -224,15 +224,14 @@ class CLIPConfig(PretrainedConfig):
outputs. Read the documentation from :class:`~transformers.PretrainedConfig` for more information.
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.
kwargs (`optional`):
Dictionary of keyword arguments. Notably:
- **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.
Dictionary of keyword arguments.
"""

model_type = "clip"
Expand Down

0 comments on commit 67d4654

Please sign in to comment.