Hugging Face's logo

Kalslice
/
distilpruned80

Text Generation
Transformers
Safetensors
gpt2
text-generation-inference
Inference Endpoints
Model card Files Community
Kalslice commited on
Commit
3dba1c4
·
verified ·
1 Parent(s): fffe935

Upload 3 files

Browse files
Files changed (2) hide show
  1. config.json +12 -57
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,5 +1,6 @@
1
  {
2
  "_name_or_path": "output",
 
3
  "activation_function": "gelu_new",
4
  "architectures": [
5
  "GPT2LMHeadModel"
@@ -8,71 +9,39 @@
8
  "bos_token_id": 50256,
9
  "embd_pdrop": 0.1,
10
  "eos_token_id": 50256,
 
 
 
11
  "initializer_range": 0.02,
 
 
 
12
  "layer_norm_epsilon": 1e-05,
13
  "model_type": "gpt2",
14
  "n_ctx": 1024,
15
  "n_embd": 768,
16
  "n_head": 12,
17
  "n_inner": null,
18
- "n_layer": 12,
19
  "n_positions": 1024,
20
  "output_attentions": true,
21
  "pruned_heads": {
22
- "3": [
23
- 0,
24
- 4,
25
- 5
26
- ],
27
- "4": [
28
- 2,
29
  4,
30
  8,
31
  10
32
  ],
33
- "5": [
34
- 0,
35
- 1,
36
- 3,
37
- 4,
38
- 5,
39
- 7,
40
- 8,
41
- 9,
42
- 11
43
- ],
44
- "6": [
45
- 1,
46
- 2,
47
- 3,
48
- 5,
49
- 6,
50
- 9,
51
- 10
52
- ],
53
- "7": [
54
  1,
55
  2,
56
  3,
57
- 4,
58
  6,
59
  7,
60
  9,
61
  10,
62
  11
63
  ],
64
- "8": [
65
- 0,
66
- 1,
67
- 2,
68
- 3,
69
- 6,
70
- 8,
71
- 9,
72
- 10,
73
- 11
74
- ],
75
- "9": [
76
  0,
77
  1,
78
  2,
@@ -84,20 +53,7 @@
84
  10,
85
  11
86
  ],
87
- "10": [
88
- 0,
89
- 1,
90
- 2,
91
- 3,
92
- 4,
93
- 5,
94
- 6,
95
- 7,
96
- 8,
97
- 10,
98
- 11
99
- ],
100
- "11": [
101
  1,
102
  2,
103
  4,
@@ -105,7 +61,6 @@
105
  6,
106
  7,
107
  9,
108
- 10,
109
  11
110
  ]
111
  },
 
1
  {
2
  "_name_or_path": "output",
3
+ "_num_labels": 1,
4
  "activation_function": "gelu_new",
5
  "architectures": [
6
  "GPT2LMHeadModel"
 
9
  "bos_token_id": 50256,
10
  "embd_pdrop": 0.1,
11
  "eos_token_id": 50256,
12
+ "id2label": {
13
+ "0": "LABEL_0"
14
+ },
15
  "initializer_range": 0.02,
16
+ "label2id": {
17
+ "LABEL_0": 0
18
+ },
19
  "layer_norm_epsilon": 1e-05,
20
  "model_type": "gpt2",
21
  "n_ctx": 1024,
22
  "n_embd": 768,
23
  "n_head": 12,
24
  "n_inner": null,
25
+ "n_layer": 6,
26
  "n_positions": 1024,
27
  "output_attentions": true,
28
  "pruned_heads": {
29
+ "2": [
 
 
 
 
 
 
30
  4,
31
  8,
32
  10
33
  ],
34
+ "3": [
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
35
  1,
36
  2,
37
  3,
 
38
  6,
39
  7,
40
  9,
41
  10,
42
  11
43
  ],
44
+ "4": [
 
 
 
 
 
 
 
 
 
 
 
45
  0,
46
  1,
47
  2,
 
53
  10,
54
  11
55
  ],
56
+ "5": [
 
 
 
 
 
 
 
 
 
 
 
 
 
57
  1,
58
  2,
59
  4,
 
61
  6,
62
  7,
63
  9,
 
64
  11
65
  ]
66
  },
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:be9602cd6a7ef97c7f10558c7deb18ecb6629c8fed85fc376904d1c11dc64705
3
- size 441882984
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c4220149cd0d228212eadde41f544fc9be7d17dd25c240c801adb50c5e1d886a
3
+ size 304829120