Remove graph opt lvls.

mfuntowicz · mfuntowicz · commit 57e369e71cec · 2021-06-29T17:41:44.000+02:00
diff --git a/src/transformers/onnx/__main__.py b/src/transformers/onnx/__main__.py
@@ -16,7 +16,6 @@
 from pathlib import Path
 from typing import Callable, Tuple, Union
 
-from onnxruntime import GraphOptimizationLevel
 from transformers.models.albert import AlbertOnnxConfig
 from transformers.models.auto import AutoTokenizer
 from transformers.models.bart import BartOnnxConfig
@@ -69,14 +68,6 @@
     "xlm-roberta": {"default": XLMRobertaOnnxConfig.default},
 }
 
-# ONNX Runtime optimization levels for humans
-ONNX_OPTIMIZATION_LEVELS = {
-    "disabled": GraphOptimizationLevel.ORT_DISABLE_ALL,
-    "default": GraphOptimizationLevel.ORT_ENABLE_BASIC,
-    "extended": GraphOptimizationLevel.ORT_ENABLE_EXTENDED,
-    "all": GraphOptimizationLevel.ORT_ENABLE_ALL,
-}
-
 
 def get_model_from_framework_and_features(framework: str, features: str, model: str):
     """
