jordiclive commited on
Commit
c920a76
·
1 Parent(s): 30131af

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +5 -1
config.json CHANGED
@@ -3,6 +3,10 @@
3
  "architectures": [
4
  "LlamaForCausalLM"
5
  ],
 
 
 
 
6
  "bos_token_id": 1,
7
  "eos_token_id": 2,
8
  "hidden_act": "silu",
@@ -25,4 +29,4 @@
25
  "use_cache": true,
26
  "use_xpos": false,
27
  "vocab_size": 32000
28
- }
 
3
  "architectures": [
4
  "LlamaForCausalLM"
5
  ],
6
+ "auto_map": {
7
+ "AutoConfig": "configuration_llama.LlamaConfig",
8
+ "AutoModelForCausalLM": "modelling_llama.LlamaForCausalLM"
9
+ },
10
  "bos_token_id": 1,
11
  "eos_token_id": 2,
12
  "hidden_act": "silu",
 
29
  "use_cache": true,
30
  "use_xpos": false,
31
  "vocab_size": 32000
32
+ }