robinzixuan commited on
Commit
3590b7f
1 Parent(s): 17dff97

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -1
config.json CHANGED
@@ -24,5 +24,10 @@
24
  "transformers_version": "4.31.0",
25
  "use_cache": true,
26
  "vocab_size": 50272,
27
- "word_embed_proj_dim": 768
 
 
 
 
 
28
  }
 
24
  "transformers_version": "4.31.0",
25
  "use_cache": true,
26
  "vocab_size": 50272,
27
+ "word_embed_proj_dim": 768,
28
+ "auto_map":
29
+ {"AutoConfig": "configuration_opt.OPTConfig",
30
+ "AutoModel": "modeling_opt.OPTModel",
31
+ "AutoModelForCausalLM": "modeling_opt.OPTForCausalLM"
32
+ }
33
  }