inclusionAI
/

Ling-lite

Text Generation

Model card Files Files and versions Community

rbao2018 commited on 11 days ago

Commit

5472194

·

1 Parent(s): 8dcffca

update config.json

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -6,7 +6,8 @@
     "auto_map": {
         "AutoConfig": "configuration_bailing_moe.BailingMoeConfig",
         "AutoModel": "modeling_bailing_moe.BailingMoeModel",
-        "AutoModelForCausalLM": "modeling_bailing_moe.BailingMoeForCausalLM"
     },
     "eos_token_id": 126081,
     "pad_token_id": 126081,

     "auto_map": {
         "AutoConfig": "configuration_bailing_moe.BailingMoeConfig",
         "AutoModel": "modeling_bailing_moe.BailingMoeModel",
+        "AutoModelForCausalLM": "modeling_bailing_moe.BailingMoeForCausalLM",
+	"AutoModelForTokenClassification": "modeling_bailing_moe.BailingMoeForTokenClassification"
     },
     "eos_token_id": 126081,
     "pad_token_id": 126081,