Skip to content
Open
Show file tree
Hide file tree
Changes from 4 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion QEfficient/exporter/export_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ def export_onnx(
input_names=input_names,
output_names=output_names,
dynamic_axes=dynamic_axes,
opset_version=13,
opset_version=17,
custom_opsets={"com.qti.aisw.onnx": 1},
)
except Exception as e:
Expand Down
3 changes: 1 addition & 2 deletions QEfficient/utils/constants.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,6 @@
ONNX_EXPORT_EXAMPLE_SEQ_LEN = 32
ONNX_EXPORT_EXAMPLE_FBS = 4
ONNX_EXPORT_EXAMPLE_NLK = 2 # Number of Logits to Keep
ONNX_EXPORT_OPSET = 13
ONNX_EXPORT_MAX_NUM_IMAGES = 1
ONNX_EXPORT_MAX_IMAGE_TILES = 4
ONNX_EXPORT_IMAGE_WIDTH = 560
Expand Down Expand Up @@ -84,7 +83,7 @@ def get_models_dir():
ONNX_EXPORT_EXAMPLE_MAX_TOP_K_IDS = 512
ONNX_EXPORT_EXAMPLE_TOP_PS = 0.80
ONNX_EXPORT_EXAMPLE_MIN_PS = 0.99
ONNX_EXPORT_OPSET = 13
ONNX_EXPORT_OPSET = 17

COMPILER = ["/opt/qti-aic/exec/qaic-exec", "-aic-hw"]
DEFAULT_AIC_HW_VERSION = "ai100"
Expand Down
Loading