tarekziade commited on
Commit
61f54c3
1 Parent(s): 6caea98

Upload 16 files

Browse files
config.json CHANGED
@@ -1,184 +1,183 @@
1
- {
2
- "_name_or_path": "mozilla/distilvit",
3
- "architectures": [
4
- "VisionEncoderDecoderModel"
5
- ],
6
- "decoder": {
7
- "_name_or_path": "distilbert/distilgpt2",
8
- "_num_labels": 1,
9
- "activation_function": "gelu_new",
10
- "add_cross_attention": true,
11
- "architectures": [
12
- "GPT2LMHeadModel"
13
- ],
14
- "attn_pdrop": 0.1,
15
- "bad_words_ids": null,
16
- "begin_suppress_tokens": null,
17
- "bos_token_id": 50256,
18
- "chunk_size_feed_forward": 0,
19
- "cross_attention_hidden_size": null,
20
- "decoder_start_token_id": null,
21
- "diversity_penalty": 0.0,
22
- "do_sample": false,
23
- "early_stopping": false,
24
- "embd_pdrop": 0.1,
25
- "encoder_no_repeat_ngram_size": 0,
26
- "eos_token_id": 50256,
27
- "exponential_decay_length_penalty": null,
28
- "finetuning_task": null,
29
- "forced_bos_token_id": null,
30
- "forced_eos_token_id": null,
31
- "id2label": {
32
- "0": "LABEL_0"
33
- },
34
- "initializer_range": 0.02,
35
- "is_decoder": true,
36
- "is_encoder_decoder": false,
37
- "label2id": {
38
- "LABEL_0": 0
39
- },
40
- "layer_norm_epsilon": 1e-05,
41
- "length_penalty": 1.0,
42
- "max_length": 20,
43
- "min_length": 0,
44
- "model_type": "gpt2",
45
- "n_ctx": 1024,
46
- "n_embd": 768,
47
- "n_head": 12,
48
- "n_inner": null,
49
- "n_layer": 6,
50
- "n_positions": 1024,
51
- "no_repeat_ngram_size": 0,
52
- "num_beam_groups": 1,
53
- "num_beams": 1,
54
- "num_return_sequences": 1,
55
- "output_attentions": false,
56
- "output_hidden_states": false,
57
- "output_scores": false,
58
- "pad_token_id": null,
59
- "prefix": null,
60
- "problem_type": null,
61
- "pruned_heads": {},
62
- "remove_invalid_values": false,
63
- "reorder_and_upcast_attn": false,
64
- "repetition_penalty": 1.0,
65
- "resid_pdrop": 0.1,
66
- "return_dict": true,
67
- "return_dict_in_generate": false,
68
- "scale_attn_by_inverse_layer_idx": false,
69
- "scale_attn_weights": true,
70
- "sep_token_id": null,
71
- "summary_activation": null,
72
- "summary_first_dropout": 0.1,
73
- "summary_proj_to_labels": true,
74
- "summary_type": "cls_index",
75
- "summary_use_proj": true,
76
- "suppress_tokens": null,
77
- "task_specific_params": {
78
- "text-generation": {
79
- "do_sample": true,
80
- "max_length": 50
81
- }
82
- },
83
- "temperature": 1.0,
84
- "tf_legacy_loss": false,
85
- "tie_encoder_decoder": false,
86
- "tie_word_embeddings": true,
87
- "tokenizer_class": null,
88
- "top_k": 50,
89
- "top_p": 1.0,
90
- "torch_dtype": null,
91
- "torchscript": false,
92
- "typical_p": 1.0,
93
- "use_bfloat16": false,
94
- "use_cache": true,
95
- "vocab_size": 50257
96
- },
97
- "decoder_start_token_id": 50256,
98
- "encoder": {
99
- "_name_or_path": "google/vit-base-patch16-224-in21k",
100
- "add_cross_attention": false,
101
- "architectures": [
102
- "ViTModel"
103
- ],
104
- "attention_probs_dropout_prob": 0.0,
105
- "bad_words_ids": null,
106
- "begin_suppress_tokens": null,
107
- "bos_token_id": null,
108
- "chunk_size_feed_forward": 0,
109
- "cross_attention_hidden_size": null,
110
- "decoder_start_token_id": null,
111
- "diversity_penalty": 0.0,
112
- "do_sample": false,
113
- "early_stopping": false,
114
- "encoder_no_repeat_ngram_size": 0,
115
- "encoder_stride": 16,
116
- "eos_token_id": null,
117
- "exponential_decay_length_penalty": null,
118
- "finetuning_task": null,
119
- "forced_bos_token_id": null,
120
- "forced_eos_token_id": null,
121
- "hidden_act": "gelu",
122
- "hidden_dropout_prob": 0.0,
123
- "hidden_size": 768,
124
- "id2label": {
125
- "0": "LABEL_0",
126
- "1": "LABEL_1"
127
- },
128
- "image_size": 224,
129
- "initializer_range": 0.02,
130
- "intermediate_size": 3072,
131
- "is_decoder": false,
132
- "is_encoder_decoder": false,
133
- "label2id": {
134
- "LABEL_0": 0,
135
- "LABEL_1": 1
136
- },
137
- "layer_norm_eps": 1e-12,
138
- "length_penalty": 1.0,
139
- "max_length": 20,
140
- "min_length": 0,
141
- "model_type": "vit",
142
- "no_repeat_ngram_size": 0,
143
- "num_attention_heads": 12,
144
- "num_beam_groups": 1,
145
- "num_beams": 1,
146
- "num_channels": 3,
147
- "num_hidden_layers": 12,
148
- "num_return_sequences": 1,
149
- "output_attentions": false,
150
- "output_hidden_states": false,
151
- "output_scores": false,
152
- "pad_token_id": null,
153
- "patch_size": 16,
154
- "prefix": null,
155
- "problem_type": null,
156
- "pruned_heads": {},
157
- "qkv_bias": true,
158
- "remove_invalid_values": false,
159
- "repetition_penalty": 1.0,
160
- "return_dict": true,
161
- "return_dict_in_generate": false,
162
- "sep_token_id": null,
163
- "suppress_tokens": null,
164
- "task_specific_params": null,
165
- "temperature": 1.0,
166
- "tf_legacy_loss": false,
167
- "tie_encoder_decoder": false,
168
- "tie_word_embeddings": true,
169
- "tokenizer_class": null,
170
- "top_k": 50,
171
- "top_p": 1.0,
172
- "torch_dtype": null,
173
- "torchscript": false,
174
- "typical_p": 1.0,
175
- "use_bfloat16": false
176
- },
177
- "eos_token_id": 50256,
178
- "is_encoder_decoder": true,
179
- "model_type": "vision-encoder-decoder",
180
- "pad_token_id": 50256,
181
- "tie_word_embeddings": false,
182
- "torch_dtype": "float32",
183
- "transformers_version": "4.40.2"
184
- }
 
1
+ {
2
+ "_name_or_path": "mozilla/distilvit",
3
+ "architectures": [
4
+ "VisionEncoderDecoderModel"
5
+ ],
6
+ "decoder": {
7
+ "_name_or_path": "distilbert/distilgpt2",
8
+ "_num_labels": 1,
9
+ "activation_function": "gelu_new",
10
+ "add_cross_attention": true,
11
+ "architectures": [
12
+ "GPT2LMHeadModel"
13
+ ],
14
+ "attn_pdrop": 0.1,
15
+ "bad_words_ids": null,
16
+ "begin_suppress_tokens": null,
17
+ "bos_token_id": 50256,
18
+ "chunk_size_feed_forward": 0,
19
+ "cross_attention_hidden_size": null,
20
+ "decoder_start_token_id": null,
21
+ "diversity_penalty": 0.0,
22
+ "do_sample": false,
23
+ "early_stopping": false,
24
+ "embd_pdrop": 0.1,
25
+ "encoder_no_repeat_ngram_size": 0,
26
+ "eos_token_id": 50256,
27
+ "exponential_decay_length_penalty": null,
28
+ "finetuning_task": null,
29
+ "forced_bos_token_id": null,
30
+ "forced_eos_token_id": null,
31
+ "id2label": {
32
+ "0": "LABEL_0"
33
+ },
34
+ "initializer_range": 0.02,
35
+ "is_decoder": true,
36
+ "is_encoder_decoder": false,
37
+ "label2id": {
38
+ "LABEL_0": 0
39
+ },
40
+ "layer_norm_epsilon": 1e-05,
41
+ "length_penalty": 1.0,
42
+ "max_length": 20,
43
+ "min_length": 0,
44
+ "model_type": "gpt2",
45
+ "n_ctx": 1024,
46
+ "n_embd": 768,
47
+ "n_head": 12,
48
+ "n_inner": null,
49
+ "n_layer": 6,
50
+ "n_positions": 1024,
51
+ "no_repeat_ngram_size": 0,
52
+ "num_beam_groups": 1,
53
+ "num_beams": 1,
54
+ "num_return_sequences": 1,
55
+ "output_attentions": false,
56
+ "output_hidden_states": false,
57
+ "output_scores": false,
58
+ "pad_token_id": null,
59
+ "prefix": null,
60
+ "problem_type": null,
61
+ "pruned_heads": {},
62
+ "remove_invalid_values": false,
63
+ "reorder_and_upcast_attn": false,
64
+ "repetition_penalty": 1.0,
65
+ "resid_pdrop": 0.1,
66
+ "return_dict": true,
67
+ "return_dict_in_generate": false,
68
+ "scale_attn_by_inverse_layer_idx": false,
69
+ "scale_attn_weights": true,
70
+ "sep_token_id": null,
71
+ "summary_activation": null,
72
+ "summary_first_dropout": 0.1,
73
+ "summary_proj_to_labels": true,
74
+ "summary_type": "cls_index",
75
+ "summary_use_proj": true,
76
+ "suppress_tokens": null,
77
+ "task_specific_params": {
78
+ "text-generation": {
79
+ "do_sample": true,
80
+ "max_length": 50
81
+ }
82
+ },
83
+ "temperature": 1.0,
84
+ "tf_legacy_loss": false,
85
+ "tie_encoder_decoder": false,
86
+ "tie_word_embeddings": true,
87
+ "tokenizer_class": null,
88
+ "top_k": 50,
89
+ "top_p": 1.0,
90
+ "torch_dtype": null,
91
+ "torchscript": false,
92
+ "typical_p": 1.0,
93
+ "use_bfloat16": false,
94
+ "use_cache": true,
95
+ "vocab_size": 50257
96
+ },
97
+ "decoder_start_token_id": 50256,
98
+ "encoder": {
99
+ "_name_or_path": "google/vit-base-patch16-224-in21k",
100
+ "add_cross_attention": false,
101
+ "architectures": [
102
+ "ViTModel"
103
+ ],
104
+ "attention_probs_dropout_prob": 0.0,
105
+ "bad_words_ids": null,
106
+ "begin_suppress_tokens": null,
107
+ "bos_token_id": null,
108
+ "chunk_size_feed_forward": 0,
109
+ "cross_attention_hidden_size": null,
110
+ "decoder_start_token_id": null,
111
+ "diversity_penalty": 0.0,
112
+ "do_sample": false,
113
+ "early_stopping": false,
114
+ "encoder_no_repeat_ngram_size": 0,
115
+ "encoder_stride": 16,
116
+ "eos_token_id": null,
117
+ "exponential_decay_length_penalty": null,
118
+ "finetuning_task": null,
119
+ "forced_bos_token_id": null,
120
+ "forced_eos_token_id": null,
121
+ "hidden_act": "gelu",
122
+ "hidden_dropout_prob": 0.0,
123
+ "hidden_size": 768,
124
+ "id2label": {
125
+ "0": "LABEL_0",
126
+ "1": "LABEL_1"
127
+ },
128
+ "image_size": 224,
129
+ "initializer_range": 0.02,
130
+ "intermediate_size": 3072,
131
+ "is_decoder": false,
132
+ "is_encoder_decoder": false,
133
+ "label2id": {
134
+ "LABEL_0": 0,
135
+ "LABEL_1": 1
136
+ },
137
+ "layer_norm_eps": 1e-12,
138
+ "length_penalty": 1.0,
139
+ "max_length": 20,
140
+ "min_length": 0,
141
+ "model_type": "vit",
142
+ "no_repeat_ngram_size": 0,
143
+ "num_attention_heads": 12,
144
+ "num_beam_groups": 1,
145
+ "num_beams": 1,
146
+ "num_channels": 3,
147
+ "num_hidden_layers": 12,
148
+ "num_return_sequences": 1,
149
+ "output_attentions": false,
150
+ "output_hidden_states": false,
151
+ "output_scores": false,
152
+ "pad_token_id": null,
153
+ "patch_size": 16,
154
+ "prefix": null,
155
+ "problem_type": null,
156
+ "pruned_heads": {},
157
+ "qkv_bias": true,
158
+ "remove_invalid_values": false,
159
+ "repetition_penalty": 1.0,
160
+ "return_dict": true,
161
+ "return_dict_in_generate": false,
162
+ "sep_token_id": null,
163
+ "suppress_tokens": null,
164
+ "task_specific_params": null,
165
+ "temperature": 1.0,
166
+ "tf_legacy_loss": false,
167
+ "tie_encoder_decoder": false,
168
+ "tie_word_embeddings": true,
169
+ "tokenizer_class": null,
170
+ "top_k": 50,
171
+ "top_p": 1.0,
172
+ "torch_dtype": null,
173
+ "torchscript": false,
174
+ "typical_p": 1.0,
175
+ "use_bfloat16": false
176
+ },
177
+ "eos_token_id": 50256,
178
+ "is_encoder_decoder": true,
179
+ "model_type": "vision-encoder-decoder",
180
+ "pad_token_id": 50256,
181
+ "tie_word_embeddings": false,
182
+ "transformers_version": "4.33.2"
183
+ }
 
generation_config.json CHANGED
@@ -1,5 +1,5 @@
1
- {
2
- "bos_token_id": 50256,
3
- "eos_token_id": 50256,
4
- "transformers_version": "4.40.2"
5
- }
 
1
+ {
2
+ "bos_token_id": 50256,
3
+ "eos_token_id": 50256,
4
+ "transformers_version": "4.33.2"
5
+ }
onnx/decoder_model.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d7025996166a391f1a16c5087f33e58b13742635f068dcfba863f481229d3aa9
3
- size 385863272
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7351a7eaeeec455e70e68007a1dcf7307e245473f53da6b13312016ad5d992d3
3
+ size 385864797
onnx/decoder_model_merged.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:67601b107ac8e081023020798926374cc1891e9b20888049b25af8b26e525036
3
- size 387339536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f951eec84e5b043b03ac6a94556e7b0f16846282bc577adec061c848119aadac
3
+ size 387342586
onnx/decoder_model_merged_quantized.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fc5f9d3406b3311eea01659346b322c08c2ca993d6c4c9a5efef56894b0cb83e
3
- size 99755449
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1ff8ee2b032955fd71b217c61751a92a6386e7b0e31228ef806a89a923bca34d
3
+ size 99759578
onnx/decoder_model_quantized.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:365eb53fde6f242e34beec14cb7b1eeb9c80538fa1527cbe3104fcbaa5bde638
3
- size 98063697
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:737759dec1b90e1e29eb61e271951db84e3e677d6f8332db64bd4a15fe931711
3
+ size 98065762
onnx/decoder_with_past_model.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f8b578768f4e7f43a54c96353cd8b06aed97c8d94411522906f021ca9195132b
3
- size 385862852
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b99196af93603df1ef7eeb9676177d99c0a4a11b01e266ae5cb9327abe968c54
3
+ size 385864377
onnx/decoder_with_past_model_quantized.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:89bab295c36f14f4dae4d6f3440f06c8620ca3b3aa671fa9e9c7114637eaa297
3
- size 98061104
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:12e5aad5d80e36cd4f9b2d45297732b01fd9e6503e6393aa393b4c13533c53e7
3
+ size 98063169
onnx/encoder_model.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fef6f2aa561891b1d39b2e79411e63cf7859ba2fa5a1a56376a13f7571c8f87a
3
- size 343440610
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aacd44f8cc26c1a0539f5db1fcb98cf3baad2c5e62583b913ed501daf1419e96
3
+ size 343440632
onnx/encoder_model_quantized.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ea7ae423fd73637b434c8a3266ea049c25b09e321cef8d0c97138ba16ab01fda
3
- size 87038151
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13062fd1801c38f1039ed439e792b9e08dc8eb0b006d4dd64146575e641c4286
3
+ size 87038173
quantize_config.json CHANGED
@@ -4,120 +4,120 @@
4
  "per_model_config": {
5
  "decoder_with_past_model": {
6
  "op_types": [
7
- "Constant",
8
  "Sqrt",
9
- "Reshape",
10
- "Concat",
11
- "Cast",
12
  "Split",
13
- "Transpose",
 
 
 
 
 
 
14
  "Add",
15
- "Slice",
 
 
16
  "Range",
 
 
 
17
  "Shape",
18
  "Squeeze",
19
- "Div",
20
- "ConstantOfShape",
21
- "Gather",
22
- "Pow",
23
- "MatMul",
24
- "Gemm",
25
  "Softmax",
26
- "Where",
27
- "Unsqueeze",
28
  "Mul",
29
- "ReduceMean",
30
- "Tanh",
31
- "Sub"
32
  ],
33
  "weight_type": "QInt8"
34
  },
35
  "decoder_model": {
36
  "op_types": [
37
- "Constant",
38
  "Sqrt",
39
- "Reshape",
40
- "Concat",
41
- "Cast",
42
  "Split",
43
- "Transpose",
 
 
 
 
 
 
44
  "Add",
45
- "Slice",
 
 
46
  "Range",
 
 
 
47
  "Shape",
48
  "Squeeze",
49
- "Div",
50
- "ConstantOfShape",
51
- "Gather",
52
- "Pow",
53
- "MatMul",
54
- "Gemm",
55
  "Softmax",
56
- "Where",
57
- "Unsqueeze",
58
  "Mul",
59
- "ReduceMean",
60
- "Tanh",
61
- "Sub"
62
  ],
63
  "weight_type": "QInt8"
64
  },
65
  "encoder_model": {
66
  "op_types": [
67
- "Constant",
68
  "Sqrt",
69
- "Reshape",
70
- "Erf",
71
- "Concat",
72
- "Transpose",
73
- "Add",
74
- "Slice",
75
  "Equal",
76
- "Shape",
77
  "Div",
78
- "ConstantOfShape",
79
- "Gather",
80
  "Pow",
81
- "MatMul",
 
 
 
 
 
 
82
  "Conv",
 
 
 
83
  "Softmax",
84
- "Where",
85
- "Unsqueeze",
86
  "Expand",
87
  "Mul",
88
- "ReduceMean",
89
- "Sub"
90
  ],
91
  "weight_type": "QUInt8"
92
  },
93
  "decoder_model_merged": {
94
  "op_types": [
95
- "Constant",
96
  "Sqrt",
97
- "Reshape",
98
- "Concat",
99
- "Cast",
100
  "Split",
101
- "Transpose",
 
 
 
 
 
 
 
102
  "Add",
103
- "Slice",
 
 
104
  "Range",
 
 
 
105
  "Shape",
106
  "Squeeze",
107
- "Div",
108
- "ConstantOfShape",
109
- "Gather",
110
- "Pow",
111
- "MatMul",
112
- "If",
113
- "Gemm",
114
  "Softmax",
115
- "Where",
116
- "Unsqueeze",
117
  "Mul",
118
- "ReduceMean",
119
- "Tanh",
120
- "Sub"
121
  ],
122
  "weight_type": "QInt8"
123
  }
 
4
  "per_model_config": {
5
  "decoder_with_past_model": {
6
  "op_types": [
 
7
  "Sqrt",
 
 
 
8
  "Split",
9
+ "Unsqueeze",
10
+ "Tanh",
11
+ "Gemm",
12
+ "MatMul",
13
+ "Cast",
14
+ "Div",
15
+ "Pow",
16
  "Add",
17
+ "Constant",
18
+ "Reshape",
19
+ "ConstantOfShape",
20
  "Range",
21
+ "Slice",
22
+ "Where",
23
+ "Concat",
24
  "Shape",
25
  "Squeeze",
26
+ "ReduceMean",
 
 
 
 
 
27
  "Softmax",
28
+ "Gather",
29
+ "Sub",
30
  "Mul",
31
+ "Transpose"
 
 
32
  ],
33
  "weight_type": "QInt8"
34
  },
35
  "decoder_model": {
36
  "op_types": [
 
37
  "Sqrt",
 
 
 
38
  "Split",
39
+ "Unsqueeze",
40
+ "Tanh",
41
+ "Gemm",
42
+ "MatMul",
43
+ "Cast",
44
+ "Div",
45
+ "Pow",
46
  "Add",
47
+ "Constant",
48
+ "Reshape",
49
+ "ConstantOfShape",
50
  "Range",
51
+ "Slice",
52
+ "Where",
53
+ "Concat",
54
  "Shape",
55
  "Squeeze",
56
+ "ReduceMean",
 
 
 
 
 
57
  "Softmax",
58
+ "Gather",
59
+ "Sub",
60
  "Mul",
61
+ "Transpose"
 
 
62
  ],
63
  "weight_type": "QInt8"
64
  },
65
  "encoder_model": {
66
  "op_types": [
 
67
  "Sqrt",
68
+ "Unsqueeze",
 
 
 
 
 
69
  "Equal",
70
+ "MatMul",
71
  "Div",
 
 
72
  "Pow",
73
+ "Add",
74
+ "Erf",
75
+ "Constant",
76
+ "Reshape",
77
+ "ConstantOfShape",
78
+ "Slice",
79
+ "Where",
80
  "Conv",
81
+ "Concat",
82
+ "Shape",
83
+ "ReduceMean",
84
  "Softmax",
85
+ "Gather",
 
86
  "Expand",
87
  "Mul",
88
+ "Sub",
89
+ "Transpose"
90
  ],
91
  "weight_type": "QUInt8"
92
  },
93
  "decoder_model_merged": {
94
  "op_types": [
 
95
  "Sqrt",
 
 
 
96
  "Split",
97
+ "If",
98
+ "Unsqueeze",
99
+ "Tanh",
100
+ "Gemm",
101
+ "MatMul",
102
+ "Cast",
103
+ "Div",
104
+ "Pow",
105
  "Add",
106
+ "Constant",
107
+ "Reshape",
108
+ "ConstantOfShape",
109
  "Range",
110
+ "Slice",
111
+ "Where",
112
+ "Concat",
113
  "Shape",
114
  "Squeeze",
115
+ "ReduceMean",
 
 
 
 
 
 
116
  "Softmax",
117
+ "Gather",
118
+ "Sub",
119
  "Mul",
120
+ "Transpose"
 
 
121
  ],
122
  "weight_type": "QInt8"
123
  }