Update README.md

README.md

@@ -12,8 +12,14 @@ from transformers import Zamba2Config, Zamba2ForCausalLM
 
 # === Step 1: Define tiny model config ===
 config = Zamba2Config(
+    num_hidden_layers=4,
+    layers_block_type=[
+        "mamba",
+        "mamba",
+        "hybrid",
+        "mamba",
+    ],
     d_model=16,
-    n_layer=46,              # Match number of Mamba/Hybrid blocks
     d_state=32,
     expand=2,
     conv_kernel=3,