Update README.md

README.md

@@ -7,6 +7,32 @@ tags: []
 
 <!-- Provide a quick summary of what the model is/does. -->
 
+```python
+from transformers import Idefics2Config, AutoProcessor, MistralConfig, Idefics2ForConditionalGeneration
+from transformers.models.idefics2.configuration_idefics2 import Idefics2VisionConfig
+
+config = Idefics2Config(
+    text_config=MistralConfig(
+        max_position_embeddings=4096 * 8,
+        vocab_size=32003,
+        hidden_size=4 * 8,
+        num_attention_heads=8,
+        intermediate_size=16,
+        num_hidden_layers=2,
+    ),
+    vision_config=Idefics2VisionConfig(
+        hidden_size=8 * 4,
+        num_attention_heads=4,
+        num_hidden_layers=2,
+        intermediate_size=16,
+    ),
+)
+model = Idefics2ForConditionalGeneration(config=config)
+processor = AutoProcessor.from_pretrained("HuggingFaceM4/idefics2-8b")
+
+model.push_to_hub("trl-internal-testing/tiny-random-idefics2")
+processor.push_to_hub("trl-internal-testing/tiny-random-idefics2")
+```
 
 
 ## Model Details