throw an error if opset < 14 is used

neuralmagic · Jun 23, 2023 · 980b6bc · 980b6bc
1 parent 68f756a
commit 980b6bc
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/integrations/clip/clip_onnx_export.py b/integrations/clip/clip_onnx_export.py
@@ -39,6 +39,11 @@ def _export_onnx(
     opset: int = 14,
     **export_kwargs,
 ):
+    # _export_onnx by default uses opset = 14 as required by CLIP and will fail
+    # for opset < 14 as certain operators are not supported.
+    if opset < 14:
+        raise ValueError("CLIP onnx export requires a minimum opset of 14")
+
     export_onnx(
         module=module,
         sample_batch=sample_batch,