Increase context window using rope base freq

config.json

@@ -16,7 +16,7 @@
   "hidden_size": 3584,
   "initializer_range": 0.02,
   "intermediate_size": 14336,
-  "max_position_embeddings": 8192,
+  "max_position_embeddings": 32768,
   "model_type": "gemma2",
   "num_attention_heads": 16,
   "num_hidden_layers": 42,
@@ -24,7 +24,7 @@
   "pad_token_id": 0,
   "query_pre_attn_scalar": 224,
   "rms_norm_eps": 1e-06,
-  "rope_theta": 10000.0,
+  "rope_theta": 160000.0,
   "sliding_window": 4096,
   "sliding_window_size": 4096,
   "torch_dtype": "bfloat16",