Skip to content

Commit 3924d7b

Browse files
JingyaHuangdacorvo
authored andcommitted
removal: skipped test
1 parent c5e654b commit 3924d7b

File tree

2 files changed

+0
-14
lines changed

2 files changed

+0
-14
lines changed

tests/exporters/test_transformers.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,6 @@
3737
from optimum.utils.testing_utils import require_sentence_transformers
3838
from parameterized import parameterized
3939
from transformers import AutoConfig, AutoModelForSeq2SeqLM, set_seed
40-
from transformers import __version__ as transformers_version
4140
from transformers.testing_utils import slow
4241

4342
from optimum.exporters.neuron import (
@@ -85,11 +84,6 @@ def _neuronx_export(
8584
dynamic_batch_size: bool = False,
8685
inline_weights_to_neff: bool = True,
8786
):
88-
# REMOVEME: convnextv2 contains a bug in the GRN layer, which is used in the convnextv2 model, but the bug has
89-
# been fixed in the transformers library on newer versions. For more info see:
90-
# https://github.com/huggingface/transformers/issues/38015
91-
if model_type == "convnextv2" and transformers_version.startswith("4.51"):
92-
self.skipTest("convnextv2 contains a bug in this version of transformers.")
9387
library_name = TasksManager.infer_library_from_model(model_name)
9488
if library_name == "sentence_transformers":
9589
model_class = TasksManager.get_model_class_for_task(task, framework="pt", library=library_name)

tests/inference/transformers/test_modeling.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -45,9 +45,6 @@
4545
PretrainedConfig,
4646
set_seed,
4747
)
48-
from transformers import (
49-
__version__ as transformers_version,
50-
)
5148

5249
from optimum.neuron import (
5350
NeuronModelForAudioClassification,
@@ -1027,11 +1024,6 @@ def test_compare_to_transformers_dyn_bs(self):
10271024
"model_arch": model_arch,
10281025
"dynamic_batch_size": True,
10291026
}
1030-
# REMOVEME: convnextv2 contains a bug in the GRN layer, which is used in the convnextv2 model, but the bug has
1031-
# been fixed in the transformers library on newer versions. For more info see:
1032-
# https://github.com/huggingface/transformers/issues/38015
1033-
if model_arch == "convnextv2" and transformers_version.startswith("4.51"):
1034-
self.skipTest("convnextv2 contains a bug in this version of transformers.")
10351027
self._setup(model_args)
10361028
self._validate_outputs(model_arch, "_dyn_bs_true", batch_size=2)
10371029

0 commit comments

Comments
 (0)