stepfun-ai
/

Step-5-Preview-BF16

Model card Files Files and versions

Update config.json

#4

by csy0225 - opened Sep 20

base: refs/heads/main

←

from: refs/pr/4

Discussion Files changed

Files changed (1) hide show

config.json +4 -7

config.json CHANGED Viewed

@@ -1,11 +1,8 @@
 {
   "architectures": [
-    "MMGPTStepRoboticsForCausalLM"
   ],
-  "auto_map": {
-    "AutoConfig": "configuration_step_robotics.StepRoboticsConfig"
-  },
-  "model_type": "step3p5v",
   "im_end_token": "<im_end>",
   "im_patch_token": "<im_patch>",
   "im_start_token": "<im_start>",
@@ -32,9 +29,9 @@
   },
   "text_config": {
     "architectures": [
-      "Step4ForCausalLM"
     ],
-    "model_type": "step4",
     "att_impl_type": "GQA",
     "torch_dtype": "bfloat16",
     "norm_dtype": "float32",

 {
   "architectures": [
+    "Step5ForConditionalGeneration"
   ],
+  "model_type": "step5_vl",
   "im_end_token": "<im_end>",
   "im_patch_token": "<im_patch>",
   "im_start_token": "<im_start>",
   },
   "text_config": {
     "architectures": [
+      "Step5ForCausalLM"
     ],
+    "model_type": "step5",
     "att_impl_type": "GQA",
     "torch_dtype": "bfloat16",
     "norm_dtype": "float32",