Update README.md

README.md

@@ -5,6 +5,15 @@ datasets:
 - HuggingFaceFW/fineweb-edu
 ---
 
-Checkpoint:
-- Model Parameters: 433.49M
-- PPL: 75
+```py
+config = {
+    "d_model": 1180,         
+    "n_rwkv_layers": 16,  
+    "n_attn_layers": 4,    
+    "n_heads": 10,
+    "seq_len": 1024,        
+    "batch_size": 4,
+    "accum_steps": 8,      
+    "lr": 4e-4,
+}
+```