Add Blip and Blip2 for pipeline tests (#21904)
* fix * add to tests * style and quality * add missing --------- Co-authored-by: NielsRogge <NielsRogge@users.noreply.github.com> Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -394,7 +394,11 @@ class BlipModelTester:
|
||||
@require_torch
|
||||
class BlipModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
|
||||
all_model_classes = (BlipModel,) if is_torch_available() else ()
|
||||
pipeline_model_mapping = {"feature-extraction": BlipModel} if is_torch_available() else {}
|
||||
pipeline_model_mapping = (
|
||||
{"feature-extraction": BlipModel, "image-to-text": BlipForConditionalGeneration}
|
||||
if is_torch_available()
|
||||
else {}
|
||||
)
|
||||
fx_compatible = False
|
||||
test_head_masking = False
|
||||
test_pruning = False
|
||||
|
||||
Reference in New Issue
Block a user