Skip to content

Commit c5e654b

Browse files
JingyaHuangdacorvo
authored andcommitted
removal: skipped test
1 parent 88db919 commit c5e654b

File tree

2 files changed

+2
-5
lines changed

2 files changed

+2
-5
lines changed

optimum/neuron/modeling_traced.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -252,11 +252,6 @@ def _from_pretrained(
252252
neuron_config=neuron_config,
253253
)
254254

255-
@classmethod
256-
def _from_transformers(cls, *args, **kwargs):
257-
# Deprecate it when optimum uses `_export` as from_pretrained_method in a stable release.
258-
return cls._export(*args, **kwargs)
259-
260255
@classmethod
261256
def _export(
262257
cls,

tests/inference/inference_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -108,6 +108,7 @@ def tearDownClass(cls):
108108

109109
class NeuronModelTestMixin(unittest.TestCase):
110110
ARCH_MODEL_MAP = {}
111+
TASK = None
111112
STATIC_INPUTS_SHAPES = {"batch_size": 1, "sequence_length": 32}
112113

113114
@classmethod
@@ -141,6 +142,7 @@ def _setup(self, model_args: Dict):
141142
model_id,
142143
**model_args,
143144
export=True,
145+
task=self.TASK,
144146
torch_dtype=torch.float32,
145147
dynamic_batch_size=dynamic_batch_size,
146148
**self.STATIC_INPUTS_SHAPES,

0 commit comments

Comments
 (0)