diff --git a/tests/ipex/test_modeling.py b/tests/ipex/test_modeling.py index 7bff39147e..38e85ed675 100644 --- a/tests/ipex/test_modeling.py +++ b/tests/ipex/test_modeling.py @@ -754,6 +754,7 @@ def test_huggingface_pipeline_streaming(self, model_arch): pipeline_kwargs={"max_new_tokens": 10}, backend="ipex", ) + self.assertIsInstance(hf_pipe.pipeline.model, IPEXModel) generator = hf_pipe.stream("Q: How do you say 'hello' in German? A:'", stop=["."]) diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py index 57d4b64764..7675b7c1e2 100644 --- a/tests/openvino/test_modeling.py +++ b/tests/openvino/test_modeling.py @@ -2833,6 +2833,7 @@ def test_huggingface_pipeline_streaming(self, model_arch): pipeline_kwargs={"max_new_tokens": 10}, backend="openvino", ) + self.assertIsInstance(hf_pipe.pipeline.model, OVBaseModel) generator = hf_pipe.stream("Q: How do you say 'hello' in German? A:'", stop=["."])