Update config.json

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "TuKoResearch/AuriStream1B_40Pred_librilight_500k",
+  "_name_or_path": "TuKoResearch/AuriStream200M_80Pred_librilight_200k",
   "architectures": ["AuriStream"],
   "model_type": "AuriStream.AuriStream",
   "auto_map": {
@@ -8,12 +8,12 @@
     "AutoModelForCausalLM": "modeling_auristream.AuriStream"
   },
   "vocab_size": 8192,
-  "n_embd": 768,
-  "n_layer": 12,
-  "n_head": 12,
+  "n_embd": 1024,
+  "n_layer": 16,
+  "n_head": 16,
   "dropout": 0.0,
   "bias": false,
   "use_rope": false,
-  "n_pred_steps": 40,
+  "n_pred_steps": 80,
   "seq_len": 4096
 }