Update config.json

config.json

@@ -1,20 +1,18 @@
 {
   "_name_or_path": "adowu/astral-128k",
-  "architectures": [
-    "MistralForCausalLM"
-  ],
+  "architectures": ["MistralForCausalLM"],
   "attention_dropout": 0.1,
   "bos_token_id": 1,
   "eos_token_id": 2,
   "hidden_act": "silu",
-  "hidden_size": 4096,
+  "hidden_size": 4096,  // Zmienione z 1024 na 4096
   "initializer_range": 0.02,
-  "intermediate_size": 8192,
+  "intermediate_size": 8192,  // Może wymagać dostosowania w zależności od modelu
   "max_position_embeddings": 128000,
   "model_type": "mistral",
-  "num_attention_heads": 8,
-  "num_hidden_layers": 16,
-  "num_key_value_heads": 8,
+  "num_attention_heads": 32,  // Zmienione z 8 na 32
+  "num_hidden_layers": 16,  // Dostosuj zgodnie z wymaganiami checkpointu
+  "num_key_value_heads": 8,  // Zazwyczaj związane z num_attention_heads
   "pad_token_id": 2,
   "rms_norm_eps": 1e-05,
   "rope_theta": 10000.0,
@@ -23,5 +21,5 @@
   "torch_dtype": "bfloat16",
   "transformers_version": "4.39.3",
   "use_cache": true,
-  "vocab_size": 32002
-}
+  "vocab_size": 32002  // Upewnij się, że jest zgodny z checkpointem
+}