From 326884e98c7f032e4bb49941ead7570ef36fb3c1 Mon Sep 17 00:00:00 2001 From: Isaac Chung Date: Fri, 20 Oct 2023 20:13:24 +0300 Subject: [PATCH] opset=18 --- optimum/exporters/onnx/model_configs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optimum/exporters/onnx/model_configs.py b/optimum/exporters/onnx/model_configs.py index 0a1d85fa34..12455fb303 100644 --- a/optimum/exporters/onnx/model_configs.py +++ b/optimum/exporters/onnx/model_configs.py @@ -752,6 +752,7 @@ def outputs(self) -> Dict[str, Dict[int, str]]: class OpenCLIPOnnxConfig(CLIPOnnxConfig): + DEFAULT_ONNX_OPSET = 18 @property def inputs(self) -> Dict[str, Dict[int, str]]: @@ -764,7 +765,6 @@ def inputs(self) -> Dict[str, Dict[int, str]]: @property def outputs(self) -> Dict[str, Dict[int, str]]: return { - "logits_scale": {0: "image_batch_size"}, "text_features": {0: "text_batch_size"}, "image_features": {0: "image_batch_size"}, }