File tree Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -1106,6 +1106,7 @@ class CLIPNormalizedConfig(NormalizedTextAndVisionConfig):
1106
1106
1107
1107
class CLIPVisionModelOnnxConfig (VisionOnnxConfig ):
1108
1108
NORMALIZED_CONFIG_CLASS = NormalizedVisionConfig
1109
+ DEFAULT_ONNX_OPSET = 14 # scaled_dot_product_attention support was added in opset 14
1109
1110
1110
1111
@property
1111
1112
def inputs (self ) -> Dict [str , Dict [int , str ]]:
@@ -1129,6 +1130,7 @@ def patch_model_for_export(
1129
1130
1130
1131
class CLIPOnnxConfig (TextAndVisionOnnxConfig ):
1131
1132
NORMALIZED_CONFIG_CLASS = CLIPNormalizedConfig
1133
+ DEFAULT_ONNX_OPSET = 14 # scaled_dot_product_attention support was added in opset 14
1132
1134
1133
1135
@property
1134
1136
def inputs (self ) -> Dict [str , Dict [int , str ]]:
You can’t perform that action at this time.
0 commit comments