File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 1818from onnx import external_data_helper
1919
2020from QEfficient .base .onnx_transforms import FP16ClipTransform
21+ from QEfficient .utils import constants
2122
2223
2324def export_onnx (
@@ -97,7 +98,7 @@ def export_onnx(
9798 input_names = input_names ,
9899 output_names = output_names ,
99100 dynamic_axes = dynamic_axes ,
100- opset_version = 13 ,
101+ opset_version = constants . ONNX_EXPORT_OPSET ,
101102 custom_opsets = {"com.qti.aisw.onnx" : 1 },
102103 )
103104 except Exception as e :
Original file line number Diff line number Diff line change 1717ONNX_EXPORT_EXAMPLE_SEQ_LEN = 32
1818ONNX_EXPORT_EXAMPLE_FBS = 4
1919ONNX_EXPORT_EXAMPLE_NLK = 2 # Number of Logits to Keep
20- ONNX_EXPORT_OPSET = 13
2120ONNX_EXPORT_MAX_NUM_IMAGES = 1
2221ONNX_EXPORT_MAX_IMAGE_TILES = 4
2322ONNX_EXPORT_IMAGE_WIDTH = 560
@@ -84,7 +83,7 @@ def get_models_dir():
8483ONNX_EXPORT_EXAMPLE_MAX_TOP_K_IDS = 512
8584ONNX_EXPORT_EXAMPLE_TOP_PS = 0.80
8685ONNX_EXPORT_EXAMPLE_MIN_PS = 0.99
87- ONNX_EXPORT_OPSET = 13
86+ ONNX_EXPORT_OPSET = 17
8887
8988COMPILER = ["/opt/qti-aic/exec/qaic-exec" , "-aic-hw" ]
9089DEFAULT_AIC_HW_VERSION = "ai100"
You can’t perform that action at this time.
0 commit comments