openbmb
/

cpm-bee-1b

Text Generation

feature-extraction

Model card Files Files and versions Community

jeffreygo commited on Jun 13, 2023

Commit

1a8d9c8

•

1 Parent(s): 0e55197

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -8,7 +8,7 @@
       "AutoConfig": "configuration_cpmbee.CpmBeeConfig",
       "AutoTokenizer": "tokenization_cpmbee.CpmBeeTokenizer",
       "AutoModel": "modeling_cpmbee.CpmBeeForCausalLM",
-      "AutoModelForCausalLM": "modeling_cpmbee.CpmBeeForCausalLM",
     },
     "vocab_size": 86583,
     "hidden_size": 4096,

       "AutoConfig": "configuration_cpmbee.CpmBeeConfig",
       "AutoTokenizer": "tokenization_cpmbee.CpmBeeTokenizer",
       "AutoModel": "modeling_cpmbee.CpmBeeForCausalLM",
+      "AutoModelForCausalLM": "modeling_cpmbee.CpmBeeForCausalLM"
     },
     "vocab_size": 86583,
     "hidden_size": 4096,