michaelbzhu
/

test-3.2B-base

custom-mbz-test

model_hub_mixin

pytorch_model_hub_mixin

Model card Files Files and versions

michaelbzhu commited on Aug 31, 2025

Commit

5051265

·

verified ·

1 Parent(s): bd9abb9

Update config.json

Files changed (1) hide show

config.json +3 -3

config.json CHANGED Viewed

@@ -3,9 +3,9 @@
     "CustomModel"
   ],
   "auto_map": {
-    "AutoConfig": "configuration_custom-mbz-test.CustomConfig",
-    "AutoModelForCausalLM": "modeling_custom-mbz-test.CustomModel",
-    "AutoModel": "modeling_custom-mbz-test.CustomModel"
   },
   "d_head": 128,
   "d_model": 8192,

     "CustomModel"
   ],
   "auto_map": {
+    "AutoConfig": "configuration_custom_mbz_test.CustomConfig",
+    "AutoModelForCausalLM": "modeling_custom_mbz_test.CustomModelForCausalLM",
+    "AutoModel": "modeling_custom_mbz_test.CustomModelForCausalLM"
   },
   "d_head": 128,
   "d_model": 8192,