Update README.md

README.md

@@ -11,6 +11,26 @@ tags: []
 
 ## Model Details
 
+### Code to generate the model
+
+```py
+from transformers import GraniteMoeConfig, GraniteMoeForCausalLM, AutoTokenizer
+
+model = GraniteMoeForCausalLM(GraniteMoeConfig(
+    hidden_size=32,
+    intermediate_size=64,
+    num_attention_heads=4,
+    num_hidden_layers=2,
+    num_key_value_heads=4,
+    pad_token_id=0,
+    vocab_size=49155,
+))
+tokenizer = AutoTokenizer.from_pretrained('ibm-granite/granite-3.0-1b-a400m-instruct')
+
+model.push_to_hub('hf-internal-testing/tiny-random-GraniteMoeForCausalLM')
+tokenizer.push_to_hub('hf-internal-testing/tiny-random-GraniteMoeForCausalLM')
+```
+
 ### Model Description
 
 <!-- Provide a longer summary of what this model is. -->