Llava: fix checkpoint_doc (#32458)

fix: add new llava like model bug
huggingface · Aug 6, 2024 · 0aa8328 · 0aa8328
1 parent 37c5ca5
commit 0aa8328
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/transformers/models/llava/modeling_llava.py b/src/transformers/models/llava/modeling_llava.py
@@ -39,6 +39,9 @@
 
 _CONFIG_FOR_DOC = "LlavaConfig"
 
+# Base docstring
+_CHECKPOINT_FOR_DOC = "llava-hf/llava-1.5-7b-hf"
+
 
 @dataclass
 # Copied from transformers.models.idefics.modeling_idefics.IdeficsCausalLMOutputWithPast with Idefics->Llava