sippycoder commited on
Commit
2ca4391
1 Parent(s): db9aa77

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -4
config.json CHANGED
@@ -1,8 +1,10 @@
1
  {
2
- "_name_or_path": "NucleusAI/test-9k-fn",
3
  "architectures": [
4
- "LlamaForCausalLM"
5
  ],
 
 
 
6
  "bos_token_id": 1,
7
  "eos_token_id": 2,
8
  "hidden_act": "silu",
@@ -11,7 +13,7 @@
11
  "intermediate_size": 17920,
12
  "max_position_embeddings": 2048,
13
  "max_sequence_length": 2048,
14
- "model_type": "llama",
15
  "num_attention_heads": 52,
16
  "num_hidden_layers": 40,
17
  "num_key_value_heads": 52,
@@ -24,4 +26,4 @@
24
  "transformers_version": "4.30.2",
25
  "use_cache": true,
26
  "vocab_size": 32000
27
- }
 
1
  {
 
2
  "architectures": [
3
+ "NucleusForCausalLM"
4
  ],
5
+ "auto_map": {
6
+ "AutoModelForCausalLM": "modeling_nucleus.NucleusForCausalLM"
7
+ },
8
  "bos_token_id": 1,
9
  "eos_token_id": 2,
10
  "hidden_act": "silu",
 
13
  "intermediate_size": 17920,
14
  "max_position_embeddings": 2048,
15
  "max_sequence_length": 2048,
16
+ "model_type": "nucleus",
17
  "num_attention_heads": 52,
18
  "num_hidden_layers": 40,
19
  "num_key_value_heads": 52,
 
26
  "transformers_version": "4.30.2",
27
  "use_cache": true,
28
  "vocab_size": 32000
29
+ }