riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

Llava: fix checkpoint_doc (#32458)

Browse Source 
fix: add new llava like model bug
This commit is contained in:
Prakarsh Kaushik 2024-08-06 14:41:59 +05:30 committed by GitHub
parent 37c5ca5eb9
commit 0aa8328293
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/models/llava/modeling_llava.py
View File

@ -39,6 +39,9 @@ logger = logging.get_logger(__name__)
_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