diff --git a/tests/ci_use/XPU_45T/run_w4a8.py b/tests/ci_use/XPU_45T/run_w4a8.py index bd9a8b4cc..b5eea23e1 100644 --- a/tests/ci_use/XPU_45T/run_w4a8.py +++ b/tests/ci_use/XPU_45T/run_w4a8.py @@ -36,7 +36,7 @@ def test_w4a8(): ) print(response.choices[0].message.content) # print(base_response) - assert any(keyword in response.choices[0].message.content for keyword in ["人工智能", "文心一言"]) + assert any(keyword in response.choices[0].message.content for keyword in ["人工智能", "文心一言", "小度"]) if __name__ == "__main__":