From 35913c826e76791addbdca018a84fbcfb8eee5b6 Mon Sep 17 00:00:00 2001 From: Shengsheng Huang Date: Wed, 27 Mar 2024 01:14:19 +0800 Subject: [PATCH] update comment --- libs/community/tests/integration_tests/llms/test_bigdl_llm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/community/tests/integration_tests/llms/test_bigdl_llm.py b/libs/community/tests/integration_tests/llms/test_bigdl_llm.py index d2c109f0c9b74..8fcbc5c81c935 100644 --- a/libs/community/tests/integration_tests/llms/test_bigdl_llm.py +++ b/libs/community/tests/integration_tests/llms/test_bigdl_llm.py @@ -5,7 +5,7 @@ def test_call() -> None: - """Test valid call to ipex-llm.""" + """Test valid call to bigdl-llm.""" llm = BigdlLLM.from_model_id( model_id="lmsys/vicuna-7b-v1.5", model_kwargs={"temperature": 0, "max_length": 16, "trust_remote_code": True}, @@ -15,7 +15,7 @@ def test_call() -> None: def test_generate() -> None: - """Test valid call to ipex-llm.""" + """Test valid call to bigdl-llm.""" llm = BigdlLLM.from_model_id( model_id="lmsys/vicuna-7b-v1.5", model_kwargs={"temperature": 0, "max_length": 16, "trust_remote_code": True},