magicslabnu
/

OutEffHop-opt-350m

Text Generation

text-generation-inference

Model card Files Files and versions Community

robinzixuan commited on Jun 15

Commit

3590b7f

•

1 Parent(s): 17dff97

Update config.json

Files changed (1) hide show

config.json +6 -1

config.json CHANGED Viewed

@@ -24,5 +24,10 @@
   "transformers_version": "4.31.0",
   "use_cache": true,
   "vocab_size": 50272,
-  "word_embed_proj_dim": 768
 }

   "transformers_version": "4.31.0",
   "use_cache": true,
   "vocab_size": 50272,
+  "word_embed_proj_dim": 768,
+  "auto_map":
+      {"AutoConfig": "configuration_opt.OPTConfig",
+        "AutoModel": "modeling_opt.OPTModel",
+        "AutoModelForCausalLM": "modeling_opt.OPTForCausalLM"
+      }
 }