Update Blip2 is_pipeline_test_to_skip method signature (#34067)

Update method signature
This commit is contained in:
Pavel Iakubovskii
2024-10-10 16:32:08 +01:00
committed by GitHub
parent e7dfb917f8
commit 8363fd8346

View File

@@ -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