[idefics] small fixes (#25764)

huggingface · Aug 28, 2023 · 13aef13 · 13aef13
1 parent 6836e9d
commit 13aef13
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/src/transformers/models/idefics/__init__.py b/src/transformers/models/idefics/__init__.py
@@ -35,7 +35,6 @@
     _import_structure["modeling_idefics"] = [
         "IDEFICS_PRETRAINED_MODEL_ARCHIVE_LIST",
         "IdeficsForVisionText2Text",
-        "IdeficsGatedCrossAttentionLayer",
         "IdeficsModel",
         "IdeficsPreTrainedModel",
     ]
@@ -62,7 +61,6 @@
         from .modeling_idefics import (
             IDEFICS_PRETRAINED_MODEL_ARCHIVE_LIST,
             IdeficsForVisionText2Text,
-            IdeficsGatedCrossAttentionLayer,
             IdeficsModel,
             IdeficsPreTrainedModel,
         )

diff --git a/src/transformers/models/idefics/modeling_idefics.py b/src/transformers/models/idefics/modeling_idefics.py
@@ -1482,9 +1482,9 @@ def forward(
         Example:
 
         ```python
-        >>> from transformers import AutoTokenizer, LlamaForCausalLM
+        >>> from transformers import AutoTokenizer, IdeficsForVisionText2Text
 
-        >>> model = LlamaForCausalLM.from_pretrained(PATH_TO_CONVERTED_WEIGHTS)
+        >>> model = IdeficsForVisionText2Text.from_pretrained(PATH_TO_CONVERTED_WEIGHTS)
         >>> tokenizer = AutoTokenizer.from_pretrained(PATH_TO_CONVERTED_TOKENIZER)
 
         >>> prompt = "Hey, are you consciours? Can you talk to me?"