From edf12bbbf2e0cd39090dfef42014e914e212700f Mon Sep 17 00:00:00 2001 From: jiqing-feng Date: Thu, 4 Jul 2024 03:44:23 -0400 Subject: [PATCH] fix name --- tests/ipex/test_modeling.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/ipex/test_modeling.py b/tests/ipex/test_modeling.py index f4c91bed0e..7ad19cada3 100644 --- a/tests/ipex/test_modeling.py +++ b/tests/ipex/test_modeling.py @@ -474,10 +474,10 @@ def test_pipeline(self, model_arch): self.assertTrue(isinstance(outputs[0]["label"], str)) def test_patched_model(self): - ipex_model = IPEXModelForQuestionAnswering.from_pretrained( + ipex_model = IPEXModelForImageClassification.from_pretrained( "Jiqing/patched_tiny_random_vit_for_image_classification" ) - transformers_model = AutoModelForQuestionAnswering.from_pretrained("hf-internal-testing/tiny-random-vit") + transformers_model = self.IPEX_MODEL_CLASS.from_pretrained("hf-internal-testing/tiny-random-vit") preprocessor = AutoFeatureExtractor.from_pretrained("hf-internal-testing/tiny-random-vit") url = "http://images.cocodataset.org/val2017/000000039769.jpg" image = Image.open(requests.get(url, stream=True).raw)