Alignment-Lab-AI commited on
Commit
6a4431d
1 Parent(s): 79a85ed

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -0
config.json CHANGED
@@ -17,6 +17,11 @@
17
  "num_key_value_heads": 8,
18
  "rms_norm_eps": 1e-06,
19
  "rope_theta": 1000000.0,
 
 
 
 
 
20
  "sliding_window": 131072,
21
  "tie_word_embeddings": false,
22
  "torch_dtype": "bfloat16",
@@ -24,4 +29,5 @@
24
  "use_cache": true,
25
  "use_sliding_window": false,
26
  "vocab_size": 152064
 
27
  }
 
17
  "num_key_value_heads": 8,
18
  "rms_norm_eps": 1e-06,
19
  "rope_theta": 1000000.0,
20
+ "rope_scaling": {
21
+ "factor": 4.0,
22
+ "original_max_position_embeddings": 32768,
23
+ "type": "yarn"
24
+ }
25
  "sliding_window": 131072,
26
  "tie_word_embeddings": false,
27
  "torch_dtype": "bfloat16",
 
29
  "use_cache": true,
30
  "use_sliding_window": false,
31
  "vocab_size": 152064
32
+
33
  }