Skip to content

Commit

Permalink
Docstring fixup
Browse files Browse the repository at this point in the history
  • Loading branch information
Rocketknight1 committed Jun 10, 2024
1 parent b1ecb64 commit c440ca5
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 2 deletions.
3 changes: 1 addition & 2 deletions src/transformers/models/idefics2/processing_idefics2.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,8 +57,7 @@ class Idefics2Processor(ProcessorMixin):
This parameter is used to build the string from the input prompt and image tokens and should match the
config.perceiver_config.resampler_n_latents value for the model used.
chat_template (`str`, *optional*): A Jinja template which will be used to convert lists of messages
in a chat into a tokenizable string. This argument is optional, and only relevant to processors that support
chat inputs.
in a chat into a tokenizable string.
"""

attributes = ["image_processor", "tokenizer"]
Expand Down
2 changes: 2 additions & 0 deletions src/transformers/models/llava/processing_llava.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,8 @@ class LlavaProcessor(ProcessorMixin):
The image processor is a required input.
tokenizer ([`LlamaTokenizerFast`], *optional*):
The tokenizer is a required input.
chat_template (`str`, *optional*): A Jinja template which will be used to convert lists of messages
in a chat into a tokenizable string.
"""

attributes = ["image_processor", "tokenizer"]
Expand Down
2 changes: 2 additions & 0 deletions src/transformers/models/llava_next/processing_llava_next.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,8 @@ class LlavaNextProcessor(ProcessorMixin):
The image processor is a required input.
tokenizer ([`LlamaTokenizerFast`], *optional*):
The tokenizer is a required input.
chat_template (`str`, *optional*): A Jinja template which will be used to convert lists of messages
in a chat into a tokenizable string.
"""

attributes = ["image_processor", "tokenizer"]
Expand Down

0 comments on commit c440ca5

Please sign in to comment.