vaibhavad
/

llama-31-mlm

Feature Extraction

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

vaibhavad commited on Sep 16, 2024

Commit

02199b4

·

verified ·

1 Parent(s): a002e17

Update config.json

Files changed (1) hide show

config.json +4 -1

config.json CHANGED Viewed

@@ -1,9 +1,12 @@
 {
   "architectures": [
-    "LlamaForCausalLM"
   ],
   "attention_bias": false,
   "attention_dropout": 0.0,
   "bos_token_id": 128000,
   "eos_token_id": [
     128001,

 {
   "architectures": [
+    "LlamaEncoderModel"
   ],
   "attention_bias": false,
   "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoModel": "vaibhavad/llama-31-mlm--modeling_llama_encoder.LlamaEncoderModel"
+  },
   "bos_token_id": 128000,
   "eos_token_id": [
     128001,