SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update Blip2 is_pipeline_test_to_skip method signature (#34067)

Browse Source 
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
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/models/blip_2/test_modeling_blip_2.py
View File

@@ -718,9 +718,16 @@ class Blip2ModelTest(ModelTesterMixin, PipelineTesterMixin, GenerationTesterMixi
# TODO: Fix the failed tests # TODO: Fix the failed tests
def is_pipeline_test_to_skip( 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'`. # Get `RuntimeError: "LayerNormKernelImpl" not implemented for 'Half'`.
return True return True