Tostino commited on
Commit
bcba660
1 Parent(s): 4cb9e1e

Fix rope scaling in config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -1
config.json CHANGED
@@ -17,7 +17,10 @@
17
  "num_key_value_heads": 40,
18
  "pretraining_tp": 1,
19
  "rms_norm_eps": 1e-05,
20
- "rope_scaling": null,
 
 
 
21
  "rope_theta": 10000.0,
22
  "tie_word_embeddings": false,
23
  "torch_dtype": "float16",
 
17
  "num_key_value_heads": 40,
18
  "pretraining_tp": 1,
19
  "rms_norm_eps": 1e-05,
20
+ "rope_scaling": {
21
+ "factor": 2.0,
22
+ "type": "linear"
23
+ },
24
  "rope_theta": 10000.0,
25
  "tie_word_embeddings": false,
26
  "torch_dtype": "float16",