ujjirox
/

yi-34b-chat

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

ujjirox commited on Dec 8, 2023

Commit

57377d6

•

1 Parent(s): 2e75ab0

update rope scaling

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -16,7 +16,7 @@
   "num_key_value_heads": 8,
   "pretraining_tp": 1,
   "rms_norm_eps": 1e-05,
-  "rope_scaling": null,
   "rope_theta": 5000000.0,
   "tie_word_embeddings": false,
   "torch_dtype": "bfloat16",

   "num_key_value_heads": 8,
   "pretraining_tp": 1,
   "rms_norm_eps": 1e-05,
+  "rope_scaling": {"type": "dynamic", "factor": 2.0},
   "rope_theta": 5000000.0,
   "tie_word_embeddings": false,
   "torch_dtype": "bfloat16",