opset=18

huggingface · Oct 20, 2023 · 326884e · 326884e
1 parent a05b4b1
commit 326884e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"},
         }