[idefics] small fixes (#25764)

src/transformers/models/idefics/__init__.py

@@ -35,7 +35,6 @@ else:
     _import_structure["modeling_idefics"] = [
         "IDEFICS_PRETRAINED_MODEL_ARCHIVE_LIST",
         "IdeficsForVisionText2Text",
-        "IdeficsGatedCrossAttentionLayer",
         "IdeficsModel",
         "IdeficsPreTrainedModel",
     ]
@@ -62,7 +61,6 @@ if TYPE_CHECKING:
         from .modeling_idefics import (
             IDEFICS_PRETRAINED_MODEL_ARCHIVE_LIST,
             IdeficsForVisionText2Text,
-            IdeficsGatedCrossAttentionLayer,
             IdeficsModel,
             IdeficsPreTrainedModel,
         )

src/transformers/models/idefics/modeling_idefics.py

@@ -1482,9 +1482,9 @@ class IdeficsForVisionText2Text(IdeficsPreTrainedModel):
         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?"