Update config.json

config.json

@@ -1,10 +1,15 @@
 {
-  "_name_or_path": "damerajee/Gaja-v1.00",
+  "_name_or_path": "damerajee/Hixtral-test",
   "architectures": [
-    "MixtralForCausalLM"
+    "HixtralForCausalLM"
   ],
   "attention_bias": false,
   "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "damerajee/Hixtral-test--configuration_Hixtral.HixtralConfig",
+    "AutoModelForCausalLM": "damerajee/Hixtral-test--modeling_Hixtral.HixtralForCausalLM",
+
+  },
   "bos_token_id": 1,
   "eos_token_id": 2,
   "hidden_act": "silu",
@@ -12,12 +17,12 @@
   "initializer_range": 0.02,
   "intermediate_size": 11008,
   "max_position_embeddings": 4096,
-  "model_type": "mixtral",
+  "model_type": "Llamoe",
   "num_attention_heads": 32,
   "num_experts_per_tok": 2,
   "num_hidden_layers": 32,
   "num_key_value_heads": 32,
-  "num_local_experts": 2,
+  "num_local_experts": 4,
   "output_router_logits": false,
   "pretraining_tp": 1,
   "rms_norm_eps": 1e-05,
@@ -27,7 +32,7 @@
   "sliding_window": null,
   "tie_word_embeddings": false,
   "torch_dtype": "float16",
-  "transformers_version": "4.39.2",
+  "transformers_version": "4.38.2",
   "use_cache": true,
   "vocab_size": 48064
-}
+}