move patching specs to onnx config

Author: echarlaix

optimum/exporters/base.py

@@ -35,8 +35,6 @@
 if TYPE_CHECKING:
     from transformers import PretrainedConfig, PreTrainedModel, TFPreTrainedModel
 
-    from .model_patcher import PatchingSpec
-
 logger = logging.get_logger(__name__)
 
 
@@ -112,7 +110,6 @@ class ExportersConfig(ABC):
     ATOL_FOR_VALIDATION: Union[float, Dict[str, float]] = 1e-5
     MIN_TORCH_VERSION = GLOBAL_MIN_TORCH_VERSION
     MIN_TRANSFORMERS_VERSION = GLOBAL_MIN_TRANSFORMERS_VERSION
-    PATCHING_SPECS: Optional[List["PatchingSpec"]] = None
     _TASK_TO_COMMON_OUTPUTS = {
         "audio-classification": ["logits"],
         "audio-frame-classification": ["logits"],

optimum/exporters/onnx/base.py

@@ -60,6 +60,8 @@
 if TYPE_CHECKING:
     from transformers import PretrainedConfig, PreTrainedModel, TFPreTrainedModel
 
+    from .model_patcher import PatchingSpec
+
     if is_diffusers_available():
         from diffusers import ModelMixin
 
@@ -102,7 +104,8 @@ class OnnxConfig(ExportersConfig):
     DEFAULT_ONNX_OPSET = 11
     VARIANTS = {"default": "The default ONNX variant."}
     DEFAULT_VARIANT = "default"
-
+    # TODO: move PATCHING_SPECS to ExportersConfig
+    PATCHING_SPECS: Optional[List["PatchingSpec"]] = None
     _TASK_TO_COMMON_OUTPUTS = {
         "audio-classification": OrderedDict({"logits": {0: "batch_size"}}),
         "audio-frame-classification": OrderedDict({"logits": {0: "batch_size", 1: "sequence_length"}}),

optimum/exporters/onnx/model_configs.py

@@ -104,7 +104,6 @@
     from transformers import PretrainedConfig
     from transformers.modeling_utils import PreTrainedModel
 
-
     if is_tf_available():
         from transformers.modeling_tf_utils import TFPreTrainedModel