Add check langchain pipeline models test

huggingface · Jan 30, 2025 · 2d2d98f · 2d2d98f
1 parent 3ef8ae2
commit 2d2d98f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -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=["."])