Skip to content

Updating transformers issue with zero-shot-image-classification #540

Open
@loadams

Description

@loadams

Updating to transformers versions beyond v4.42.4 causes issues with the CI tests in the legacy mode. The zero-shot-image-classification tests fail with:

FAILED test_non_persistent_deployment.py::test_single_GPU[None-50050-False-28080-fp16-1-False-False-1-True-False-ds_config0-zero-shot-image-classification-openai/clip-vit-base-patch32-query6-non-persistent] - ValueError: Attempting to cast a BatchFeature to type None. This is not supported.

We have isolated the problematic commit to this one: huggingface/transformers#31342

../../mii/legacy/client.py:144: in query
    return task_methods.run_inference(inference_pipeline, args, query_kwargs)
../../mii/legacy/method_table.py:255: in run_inference
    return inference_pipeline(image, candidate_labels=candidate_labels, **kwargs)
../../../venv/lib/python3.12/site-packages/transformers/pipelines/zero_shot_image_classification.py:107: in __call__
    return super().__call__(images, **kwargs)
../../../venv/lib/python3.12/site-packages/transformers/pipelines/base.py:1257: in __call__
    return self.run_single(inputs, preprocess_params, forward_params, postprocess_params)
../../../venv/lib/python3.12/site-packages/transformers/pipelines/base.py:1263: in run_single
    model_inputs = self.preprocess(inputs, **preprocess_params)
../../../venv/lib/python3.12/site-packages/transformers/pipelines/zero_shot_image_classification.py:124: in preprocess
    inputs = inputs.to(self.torch_dtype)

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions