Skip to content

GitLab

Unverified Commit 0aa83282 authored by Prakarsh Kaushik's avatar Prakarsh Kaushik Committed by GitHub
Browse files

Llava: fix checkpoint_doc (#32458) 

fix: add new llava like model bug
parent 37c5ca5e
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
src/transformers/models/llava/modeling_llava.py
View file @ 0aa83282
...@@ -39,6 +39,9 @@ logger = logging.get_logger(__name__) ...@@ -39,6 +39,9 @@ logger = logging.get_logger(__name__)
_CONFIG_FOR_DOC = "LlavaConfig" _CONFIG_FOR_DOC = "LlavaConfig"
# Base docstring
_CHECKPOINT_FOR_DOC = "llava-hf/llava-1.5-7b-hf"
@dataclass @dataclass
# Copied from transformers.models.idefics.modeling_idefics.IdeficsCausalLMOutputWithPast with Idefics->Llava # Copied from transformers.models.idefics.modeling_idefics.IdeficsCausalLMOutputWithPast with Idefics->Llava
......
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