Update README.md

README.md

@@ -1,3 +1,39 @@
----
-license: apache-2.0
----
+---
+license: apache-2.0
+---
+
+```python
+import os
+from transformers import (
+    AutoConfig,
+    AutoModelForImageTextToText,
+    AutoProcessor,
+    AutoTokenizer,
+)
+
+model_id = "mistralai/Mistral-Small-3.1-24B-Instruct-2503"
+config = AutoConfig.from_pretrained(model_id)
+
+config.text_config.num_hidden_layers = 2
+config.text_config.hidden_size = 8
+config.text_config.intermediate_size = 64
+config.text_config.num_attention_heads = 8
+config.text_config.num_key_value_heads = 4
+config.text_config.head_dim = 32
+
+config.vision_config.num_hidden_layers = 2
+config.vision_config.hidden_size = 128
+config.vision_config.intermediate_size = 64
+config.vision_config.num_attention_heads = 4
+config.vision_config.head_dim = 32
+
+model = AutoModelForImageTextToText.from_config(config)
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+processor = AutoProcessor.from_pretrained(model_id)
+
+output_dir = "./tiny-random-mistral3"
+os.makedirs(output_dir, exist_ok=True)
+model.save_pretrained(output_dir)
+tokenizer.save_pretrained(output_dir)
+processor.save_pretrained(output_dir)
+```