lemms commited on
Commit
946abb0
·
verified ·
1 Parent(s): e69e071

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +10 -8
config.json CHANGED
@@ -1,15 +1,17 @@
1
  {
2
  "architectures": [
3
- "GPTModel"
4
  ],
5
- "model_type": "gpt",
6
  "vocab_size": 32000,
7
- "n_layer": 6,
8
- "n_head": 8,
9
- "n_embd": 512,
10
- "block_size": 1024,
11
- "dropout": 0.1,
12
- "bias": true,
 
 
13
  "torch_dtype": "float32",
14
  "transformers_version": "4.0.0",
15
  "openllm_version": "0.1.0",
 
1
  {
2
  "architectures": [
3
+ "LlamaForCausalLM"
4
  ],
5
+ "model_type": "llama",
6
  "vocab_size": 32000,
7
+ "hidden_size": 512,
8
+ "intermediate_size": 2048,
9
+ "num_hidden_layers": 6,
10
+ "num_attention_heads": 8,
11
+ "max_position_embeddings": 1024,
12
+ "rms_norm_eps": 1e-6,
13
+ "rope_theta": 10000.0,
14
+ "use_cache": true,
15
  "torch_dtype": "float32",
16
  "transformers_version": "4.0.0",
17
  "openllm_version": "0.1.0",