Update config.json

config.json

@@ -1,26 +1,17 @@
-from transformers import PretrainedConfig
-
-class AuriStreamConfig(PretrainedConfig):
-    model_type = "AuriStream.AuriStream"
-
-    def __init__(
-        self,
-        n_layer=32,
-        n_head=32,
-        n_embd=4096,
-        vocab_size=8192,
-        dropout=0.0,
-        bias=False,
-        use_rope=True,
-        n_pred_steps=20,
-        **kwargs
-    ):
-        self.n_layer = n_layer
-        self.n_head = n_head
-        self.n_embd = n_embd
-        self.vocab_size = vocab_size
-        self.dropout = dropout
-        self.bias = bias
-        self.use_rope = use_rope
-        self.n_pred_steps = n_pred_steps
-        super().__init__(**kwargs)
+{
+  "_name_or_path": "TuKoResearch/AuriStream7B20Pred",
+  "architectures": ["AuriStream"],
+  "model_type": "AuriStream.AuriStream",
+  "auto_map": {
+    "AutoConfig": "configuration_auristream.AuriStreamConfig",
+    "AutoModelForCausalLM": "modeling_auristream.AuriStream"
+  },
+  "vocab_size": 8192,
+  "n_embd": 4096,
+  "n_layer": 32,
+  "n_head": 32,
+  "dropout": 0.0,
+  "bias": false,
+  "use_rope": true,
+  "n_pred_steps": 20
+}