MBZUAI-LLM
/

SlimPajama-DC

Text Generation

dataset combination

Model card Files Files and versions

Jason0214 commited on Apr 14, 2024

Commit

ced2d75

·

verified ·

1 Parent(s): 56ff6ea

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -5,7 +5,7 @@
     ],
     "auto_map": {
         "AutoConfig": "MBZUAI-LLM/SlimPajama-DC--configuration_btlm.BTLMConfig",
-        "AutoModelForCausalLM": "MBZUAI-LLM/SlimPajama-DC--modeling_btlm.BTLMModel",
         "AutoModelForCausalLM": "MBZUAI-LLM/SlimPajama-DC--modeling_btlm.BTLMLMHeadModel"
     },
     "model_type": "btlm",

     ],
     "auto_map": {
         "AutoConfig": "MBZUAI-LLM/SlimPajama-DC--configuration_btlm.BTLMConfig",
+        "AutoModel": "MBZUAI-LLM/SlimPajama-DC--modeling_btlm.BTLMModel",
         "AutoModelForCausalLM": "MBZUAI-LLM/SlimPajama-DC--modeling_btlm.BTLMLMHeadModel"
     },
     "model_type": "btlm",