File tree Expand file tree Collapse file tree 2 files changed +0
-14
lines changed Expand file tree Collapse file tree 2 files changed +0
-14
lines changed Original file line number Diff line number Diff line change 3737from optimum .utils .testing_utils import require_sentence_transformers
3838from parameterized import parameterized
3939from transformers import AutoConfig , AutoModelForSeq2SeqLM , set_seed
40- from transformers import __version__ as transformers_version
4140from transformers .testing_utils import slow
4241
4342from 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 )
Original file line number Diff line number Diff line change 4545 PretrainedConfig ,
4646 set_seed ,
4747)
48- from transformers import (
49- __version__ as transformers_version ,
50- )
5148
5249from 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
You can’t perform that action at this time.
0 commit comments