From 8363fd8346e399d22d85091238e1b9eca34849db Mon Sep 17 00:00:00 2001 From: Pavel Iakubovskii Date: Thu, 10 Oct 2024 16:32:08 +0100 Subject: [PATCH] Update Blip2 `is_pipeline_test_to_skip` method signature (#34067) Update method signature --- tests/models/blip_2/test_modeling_blip_2.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/tests/models/blip_2/test_modeling_blip_2.py b/tests/models/blip_2/test_modeling_blip_2.py index cee5d710a8..f2ccb2da8d 100644 --- a/tests/models/blip_2/test_modeling_blip_2.py +++ b/tests/models/blip_2/test_modeling_blip_2.py @@ -718,9 +718,16 @@ class Blip2ModelTest(ModelTesterMixin, PipelineTesterMixin, GenerationTesterMixi # TODO: Fix the failed tests def is_pipeline_test_to_skip( - self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name + self, + pipeline_test_case_name, + config_class, + model_architecture, + tokenizer_name, + image_processor_name, + feature_extractor_name, + processor_name, ): - if pipeline_test_casse_name == "VisualQuestionAnsweringPipelineTests": + if pipeline_test_case_name == "VisualQuestionAnsweringPipelineTests": # Get `RuntimeError: "LayerNormKernelImpl" not implemented for 'Half'`. return True