memorease commited on
Commit
5377bed
·
verified ·
1 Parent(s): 1c62e2c

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -0
config.json CHANGED
@@ -1,6 +1,10 @@
1
  {
2
  "model_type": "llama",
3
  "architectures": ["LlamaForCausalLM"],
 
 
 
 
4
  "hidden_size": 2048,
5
  "intermediate_size": 5632,
6
  "max_position_embeddings": 2048,
 
1
  {
2
  "model_type": "llama",
3
  "architectures": ["LlamaForCausalLM"],
4
+ "auto_map": {
5
+ "AutoConfig": "llama.LlamaConfig",
6
+ "AutoModelForCausalLM": "llama.LlamaForCausalLM"
7
+ },
8
  "hidden_size": 2048,
9
  "intermediate_size": 5632,
10
  "max_position_embeddings": 2048,