abhinavkulkarni commited on
Commit
32b0ade
1 Parent(s): f8818fb

Upload config

Browse files
Files changed (1) hide show
  1. config.json +4 -1
config.json CHANGED
@@ -13,11 +13,14 @@
13
  "model_type": "llama",
14
  "num_attention_heads": 40,
15
  "num_hidden_layers": 40,
 
16
  "pad_token_id": 0,
 
17
  "rms_norm_eps": 1e-06,
 
18
  "tie_word_embeddings": false,
19
  "torch_dtype": "float16",
20
- "transformers_version": "4.30.0.dev0",
21
  "use_cache": true,
22
  "vocab_size": 32000
23
  }
 
13
  "model_type": "llama",
14
  "num_attention_heads": 40,
15
  "num_hidden_layers": 40,
16
+ "num_key_value_heads": 40,
17
  "pad_token_id": 0,
18
+ "pretraining_tp": 1,
19
  "rms_norm_eps": 1e-06,
20
+ "rope_scaling": null,
21
  "tie_word_embeddings": false,
22
  "torch_dtype": "float16",
23
+ "transformers_version": "4.31.0",
24
  "use_cache": true,
25
  "vocab_size": 32000
26
  }