Skip to content
This repository was archived by the owner on Nov 27, 2024. It is now read-only.

Commit 48c18a4

Browse files
committed
use onnx opset 16
1 parent c1b8fee commit 48c18a4

File tree

4 files changed

+7
-14
lines changed

4 files changed

+7
-14
lines changed

OnnxStack.Converter/stable_cascade/config_decoder.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@
5353
"convert": {
5454
"type": "OnnxConversion",
5555
"config": {
56-
"target_opset": 14,
56+
"target_opset": 16,
5757
"save_as_external_data": true,
5858
"all_tensors_to_one_file": true,
5959
"external_data_name": "weights.pb"
@@ -66,7 +66,7 @@
6666
"opt_level": 0,
6767
"float16": true,
6868
"use_gpu": true,
69-
"keep_io_types": false,
69+
"keep_io_types": true,
7070
"optimization_options": {
7171
"enable_gelu": true,
7272
"enable_layer_norm": true,

OnnxStack.Converter/stable_cascade/config_prior.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@
5454
"convert": {
5555
"type": "OnnxConversion",
5656
"config": {
57-
"target_opset": 14,
57+
"target_opset": 16,
5858
"save_as_external_data": true,
5959
"all_tensors_to_one_file": true,
6060
"external_data_name": "weights.pb"
@@ -67,7 +67,7 @@
6767
"opt_level": 0,
6868
"float16": true,
6969
"use_gpu": true,
70-
"keep_io_types": false,
70+
"keep_io_types": true,
7171
"optimization_options": {
7272
"enable_gelu": true,
7373
"enable_layer_norm": true,

OnnxStack.Converter/stable_cascade/config_text_encoder.json

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -48,15 +48,7 @@
4848
"convert": {
4949
"type": "OnnxConversion",
5050
"config": {
51-
"target_opset": 14
52-
}
53-
},
54-
"ov_convert": {
55-
"type": "OpenVINOConversion",
56-
"config": {
57-
"user_script": "models.py",
58-
"example_input_func": "text_encoder_conversion_inputs",
59-
"output_model": "text_encoder"
51+
"target_opset": 16
6052
}
6153
},
6254
"optimize": {
@@ -66,7 +58,7 @@
6658
"opt_level": 0,
6759
"float16": true,
6860
"use_gpu": true,
69-
"keep_io_types": false,
61+
"keep_io_types": true,
7062
"optimization_options": {
7163
"enable_gelu": true,
7264
"enable_layer_norm": true,
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,3 @@
11
/footprints/
2+
/cache/
23
/result_*.png

0 commit comments

Comments
 (0)