robertmyers commited on
Commit
1f3dce7
·
1 Parent(s): 794727c

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +49 -19
config.json CHANGED
@@ -1,26 +1,57 @@
1
  {
2
- "_name_or_path": "EleutherAI/gpt-j-6B",
3
  "activation_function": "gelu_new",
4
  "architectures": [
5
- "GPTJForCausalLM"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  ],
7
- "attn_pdrop": 0.0,
8
  "bos_token_id": 50256,
9
- "embd_pdrop": 0.0,
10
  "eos_token_id": 50256,
11
  "gradient_checkpointing": false,
 
12
  "initializer_range": 0.02,
 
13
  "layer_norm_epsilon": 1e-05,
14
- "model_type": "gptj",
15
- "n_embd": 4096,
16
- "n_head": 16,
17
- "n_inner": null,
18
- "n_layer": 28,
19
- "n_positions": 2048,
20
- "resid_pdrop": 0.0,
21
- "rotary": true,
22
- "rotary_dim": 64,
23
- "scale_attn_weights": true,
24
  "summary_activation": null,
25
  "summary_first_dropout": 0.1,
26
  "summary_proj_to_labels": true,
@@ -30,13 +61,12 @@
30
  "text-generation": {
31
  "do_sample": true,
32
  "max_length": 50,
33
- "temperature": 1.0
34
  }
35
  },
36
- "tie_word_embeddings": false,
37
  "tokenizer_class": "GPT2Tokenizer",
38
- "torch_dtype": "float32",
39
- "transformers_version": "4.20.1",
40
  "use_cache": true,
41
- "vocab_size": 50400
 
42
  }
 
1
  {
 
2
  "activation_function": "gelu_new",
3
  "architectures": [
4
+ "GPTNeoForCausalLM"
5
+ ],
6
+ "attention_dropout": 0,
7
+ "attention_layers": [
8
+ "global",
9
+ "local",
10
+ "global",
11
+ "local",
12
+ "global",
13
+ "local",
14
+ "global",
15
+ "local",
16
+ "global",
17
+ "local",
18
+ "global",
19
+ "local",
20
+ "global",
21
+ "local",
22
+ "global",
23
+ "local",
24
+ "global",
25
+ "local",
26
+ "global",
27
+ "local",
28
+ "global",
29
+ "local",
30
+ "global",
31
+ "local"
32
+ ],
33
+ "attention_types": [
34
+ [
35
+ [
36
+ "global",
37
+ "local"
38
+ ],
39
+ 12
40
+ ]
41
  ],
 
42
  "bos_token_id": 50256,
43
+ "embed_dropout": 0,
44
  "eos_token_id": 50256,
45
  "gradient_checkpointing": false,
46
+ "hidden_size": 2048,
47
  "initializer_range": 0.02,
48
+ "intermediate_size": null,
49
  "layer_norm_epsilon": 1e-05,
50
+ "max_position_embeddings": 2048,
51
+ "model_type": "gpt_neo",
52
+ "num_heads": 16,
53
+ "num_layers": 24,
54
+ "resid_dropout": 0,
 
 
 
 
 
55
  "summary_activation": null,
56
  "summary_first_dropout": 0.1,
57
  "summary_proj_to_labels": true,
 
61
  "text-generation": {
62
  "do_sample": true,
63
  "max_length": 50,
64
+ "temperature": 0.9
65
  }
66
  },
 
67
  "tokenizer_class": "GPT2Tokenizer",
68
+ "transformers_version": "4.9.0.dev0",
 
69
  "use_cache": true,
70
+ "vocab_size": 50257,
71
+ "window_size": 256
72
  }