add depreciation warning

huggingface · Nov 2, 2023 · 6c4e2fc · 6c4e2fc
1 parent 0665e08
commit 6c4e2fc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/optimum/onnxruntime/base.py b/optimum/onnxruntime/base.py
@@ -14,11 +14,11 @@
 """Defines the base classes that are used to perform inference with ONNX Runtime of Transformers models."""
 
 from abc import abstractmethod
-from typing import TYPE_CHECKING, Dict, List, Optional, Set, Tuple, Union
+from typing import TYPE_CHECKING, Dict, Optional, Set, Tuple, Union
 
 import numpy as np
 import torch
-from transformers.modeling_outputs import BaseModelOutput, CausalLMOutputWithCrossAttentions, Seq2SeqLMOutput
+from transformers.modeling_outputs import BaseModelOutput, Seq2SeqLMOutput
 
 from onnxruntime import InferenceSession
 
@@ -495,5 +495,9 @@ def prepare_inputs_for_merged(
 
 
 class ORTDecoder(ORTDecoderForSeq2Seq):
-    # TODO : add warning message
-    pass
+
+    def __init__(self, *args, **kwargs):
+        logger.warning(
+            f"The class `ORTDecoder` is deprecated and will be removed in optimum v1.15.0, please use `ORTDecoderForSeq2Seq` instead."
+        )
+        super().__init__(*args, **kwargs)
diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py
@@ -91,7 +91,7 @@
     ORTModelForVision2Seq,
     ORTStableDiffusionPipeline,
 )
-from optimum.onnxruntime.base import ORTDecoder, ORTDecoderForSeq2Seq, ORTEncoder
+from optimum.onnxruntime.base import ORTDecoderForSeq2Seq, ORTEncoder
 from optimum.onnxruntime.modeling_diffusion import (
     ORTModelTextEncoder,
     ORTModelUnet,