allenai
/

Flex-reddit-2x7B-1T

Text Generation

Mixture of Experts

Model card Files Files and versions

shanearora commited on 23 days ago

Commit

a4d7bb1

·

verified ·

1 Parent(s): 5ae37b0

Update to flex_olmo model type

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "Olmoe2ForCausalLM"
   ],
   "attention_bias": false,
   "attention_dropout": 0.0,
@@ -11,7 +11,7 @@
   "initializer_range": 0.02,
   "intermediate_size": 11008,
   "max_position_embeddings": 4096,
-  "model_type": "olmoe2",
   "norm_topk_prob": false,
   "num_attention_heads": 32,
   "num_experts": 2,

 {
   "architectures": [
+    "FlexOlmoForCausalLM"
   ],
   "attention_bias": false,
   "attention_dropout": 0.0,
   "initializer_range": 0.02,
   "intermediate_size": 11008,
   "max_position_embeddings": 4096,
+  "model_type": "flex_olmo",
   "norm_topk_prob": false,
   "num_attention_heads": 32,
   "num_experts": 2,