01-ai
/

Yi-6B

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

cArlIcon commited on Nov 2, 2023

Commit

672a9fb

•

1 Parent(s): 57eafec

update config

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -13,7 +13,7 @@
   "hidden_size": 4096,
   "initializer_range": 0.02,
   "intermediate_size": 11008,
-  "max_position_embeddings": 200000,
   "model_type": "Yi",
   "num_attention_heads": 32,
   "num_hidden_layers": 32,

   "hidden_size": 4096,
   "initializer_range": 0.02,
   "intermediate_size": 11008,
+  "max_position_embeddings": 4096,
   "model_type": "Yi",
   "num_attention_heads": 32,
   "num_hidden_layers": 32,