Update cognitivess_model/configuration_cognitivess.py

cognitivess_model/configuration_cognitivess.py

@@ -110,23 +110,23 @@ class CognitivessConfig(PretrainedConfig):
 
     def __init__(
         self,
-        vocab_size=32000,
+        vocab_size=128256,
         hidden_size=4096,
-        intermediate_size=11008,
+        intermediate_size=14336,
         num_hidden_layers=32,
         num_attention_heads=32,
-        num_key_value_heads=None,
+        num_key_value_heads=8,
         hidden_act="silu",
-        max_position_embeddings=2048,
+        max_position_embeddings=8192,
         initializer_range=0.02,
-        rms_norm_eps=1e-6,
+        rms_norm_eps=1e-05,
         use_cache=True,
-        pad_token_id=None,
-        bos_token_id=1,
-        eos_token_id=2,
+        pad_token_id=0,
+        bos_token_id=128000,
+        eos_token_id=128001,
         pretraining_tp=1,
         tie_word_embeddings=False,
-        rope_theta=10000.0,
+        rope_theta=500000.0,
         rope_scaling=None,
         attention_bias=False,
         attention_dropout=0.0,