Hugging Face's logo

Xenova
/
pygmalion-350m

Text Generation
Transformers.js
ONNX
opt
Model card Files Community
Xenova HF staff commited on
Commit
f07aa2e
•
1 Parent(s): b1b9229

Upload folder using huggingface_hub

Browse files
Files changed (14) hide show
  1. config.json +30 -0
  2. generation_config.json +7 -0
  3. merges.txt +0 -0
  4. onnx/decoder_model.onnx +3 -0
  5. onnx/decoder_model_merged.onnx +3 -0
  6. onnx/decoder_model_merged_quantized.onnx +3 -0
  7. onnx/decoder_model_quantized.onnx +3 -0
  8. onnx/decoder_with_past_model.onnx +3 -0
  9. onnx/decoder_with_past_model_quantized.onnx +3 -0
  10. quantize_config.json +107 -0
  11. special_tokens_map.json +30 -0
  12. tokenizer.json +0 -0
  13. tokenizer_config.json +40 -0
  14. vocab.json +0 -0
config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "PygmalionAI/pygmalion-350m",
3
+ "_remove_final_layer_norm": false,
4
+ "activation_dropout": 0.0,
5
+ "activation_function": "relu",
6
+ "architectures": [
7
+ "OPTForCausalLM"
8
+ ],
9
+ "attention_dropout": 0.0,
10
+ "bos_token_id": 2,
11
+ "do_layer_norm_before": false,
12
+ "dropout": 0.1,
13
+ "enable_bias": true,
14
+ "eos_token_id": 2,
15
+ "ffn_dim": 4096,
16
+ "hidden_size": 1024,
17
+ "init_std": 0.02,
18
+ "layer_norm_elementwise_affine": true,
19
+ "layerdrop": 0.0,
20
+ "max_position_embeddings": 2048,
21
+ "model_type": "opt",
22
+ "num_attention_heads": 16,
23
+ "num_hidden_layers": 24,
24
+ "pad_token_id": 1,
25
+ "prefix": "</s>",
26
+ "transformers_version": "4.34.0.dev0",
27
+ "use_cache": true,
28
+ "vocab_size": 50272,
29
+ "word_embed_proj_dim": 512
30
+ }
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 2,
4
+ "eos_token_id": 2,
5
+ "pad_token_id": 1,
6
+ "transformers_version": "4.34.0.dev0"
7
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
onnx/decoder_model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:666dd7991155d341e49ff937a0ad24f46c5441e31c2d6db158bb15b827737964
3
+ size 1325437829
onnx/decoder_model_merged.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dfcec0625656dfea2c00c0ebd688a06ab5dba35aaf58ad02189a55f9bbb78368
3
+ size 1326092776
onnx/decoder_model_merged_quantized.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d4dc611308caacc909bc9b79a2174e9f6c3b3009c6c0df1b4b025dfebb094f3
3
+ size 334237611
onnx/decoder_model_quantized.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb9cb0df298d5926ab04816be81ea3135e0609297589f30ab4adc629e5ffb447
3
+ size 333197155
onnx/decoder_with_past_model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e4c126f2caa369bb7a88d505a962d16ec939b68dccb41735721561fd12352567
3
+ size 1325445450
onnx/decoder_with_past_model_quantized.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c16570f6da7f4738e8ade2693af2c61abb5b682add3076071f1aebaf5824b740
3
+ size 333205447
quantize_config.json ADDED
@@ -0,0 +1,107 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "per_channel": false,
3
+ "reduce_range": false,
4
+ "per_model_config": {
5
+ "decoder_with_past_model": {
6
+ "op_types": [
7
+ "Expand",
8
+ "Transpose",
9
+ "Max",
10
+ "Squeeze",
11
+ "Where",
12
+ "Slice",
13
+ "Equal",
14
+ "Gather",
15
+ "Reshape",
16
+ "Div",
17
+ "Concat",
18
+ "ReduceMean",
19
+ "Constant",
20
+ "Softmax",
21
+ "CumSum",
22
+ "Relu",
23
+ "Sub",
24
+ "Mul",
25
+ "Unsqueeze",
26
+ "Gemm",
27
+ "Add",
28
+ "Cast",
29
+ "MatMul",
30
+ "Sqrt",
31
+ "Pow",
32
+ "Shape",
33
+ "ConstantOfShape"
34
+ ],
35
+ "weight_type": "QInt8"
36
+ },
37
+ "decoder_model_merged": {
38
+ "op_types": [
39
+ "Expand",
40
+ "Transpose",
41
+ "Max",
42
+ "Squeeze",
43
+ "Where",
44
+ "Slice",
45
+ "Equal",
46
+ "Gather",
47
+ "Reshape",
48
+ "If",
49
+ "Div",
50
+ "Concat",
51
+ "ReduceMean",
52
+ "Constant",
53
+ "Softmax",
54
+ "CumSum",
55
+ "Relu",
56
+ "Sub",
57
+ "Mul",
58
+ "Unsqueeze",
59
+ "Gemm",
60
+ "Add",
61
+ "Range",
62
+ "Cast",
63
+ "Less",
64
+ "MatMul",
65
+ "Sqrt",
66
+ "Pow",
67
+ "Shape",
68
+ "ConstantOfShape"
69
+ ],
70
+ "weight_type": "QInt8"
71
+ },
72
+ "decoder_model": {
73
+ "op_types": [
74
+ "Expand",
75
+ "Transpose",
76
+ "Max",
77
+ "Squeeze",
78
+ "Where",
79
+ "Slice",
80
+ "Equal",
81
+ "Gather",
82
+ "Reshape",
83
+ "Div",
84
+ "Concat",
85
+ "ReduceMean",
86
+ "Constant",
87
+ "Softmax",
88
+ "CumSum",
89
+ "Relu",
90
+ "Sub",
91
+ "Mul",
92
+ "Unsqueeze",
93
+ "Gemm",
94
+ "Add",
95
+ "Range",
96
+ "Cast",
97
+ "Less",
98
+ "MatMul",
99
+ "Sqrt",
100
+ "Pow",
101
+ "Shape",
102
+ "ConstantOfShape"
103
+ ],
104
+ "weight_type": "QInt8"
105
+ }
106
+ }
107
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "</s>",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "</s>",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<pad>",
18
+ "lstrip": false,
19
+ "normalized": true,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "unk_token": {
24
+ "content": "</s>",
25
+ "lstrip": false,
26
+ "normalized": true,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ }
30
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "add_prefix_space": false,
4
+ "bos_token": {
5
+ "__type": "AddedToken",
6
+ "content": "</s>",
7
+ "lstrip": false,
8
+ "normalized": true,
9
+ "rstrip": false,
10
+ "single_word": false
11
+ },
12
+ "clean_up_tokenization_spaces": true,
13
+ "eos_token": {
14
+ "__type": "AddedToken",
15
+ "content": "</s>",
16
+ "lstrip": false,
17
+ "normalized": true,
18
+ "rstrip": false,
19
+ "single_word": false
20
+ },
21
+ "errors": "replace",
22
+ "model_max_length": 1000000000000000019884624838656,
23
+ "pad_token": {
24
+ "__type": "AddedToken",
25
+ "content": "<pad>",
26
+ "lstrip": false,
27
+ "normalized": true,
28
+ "rstrip": false,
29
+ "single_word": false
30
+ },
31
+ "tokenizer_class": "GPT2Tokenizer",
32
+ "unk_token": {
33
+ "__type": "AddedToken",
34
+ "content": "</s>",
35
+ "lstrip": false,
36
+ "normalized": true,
37
+ "rstrip": false,
38
+ "single_word": false
39
+ }
40
+ }
vocab.json ADDED
The diff for this file is too large to render. See raw diff