From 2d2d98f03333dec4112280810cc02d2ce18e6e2f Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Thu, 30 Jan 2025 11:31:11 +0100 Subject: [PATCH] Add check langchain pipeline models test --- tests/ipex/test_modeling.py | 1 + tests/openvino/test_modeling.py | 1 + 2 files changed, 2 insertions(+) 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=["."])