From 4be54ad2370a81ec36fbba03dadea4febaa68423 Mon Sep 17 00:00:00 2001 From: AllentDan Date: Thu, 7 Mar 2024 10:03:34 +0800 Subject: [PATCH] fix UT --- tests/test_lmdeploy/test_model.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/test_lmdeploy/test_model.py b/tests/test_lmdeploy/test_model.py index e4319f7a56..5f1f879d8e 100644 --- a/tests/test_lmdeploy/test_model.py +++ b/tests/test_lmdeploy/test_model.py @@ -106,7 +106,6 @@ def test_baichuan(): assert model.get_prompt(prompt, sequence_start=True) == prompt assert model.get_prompt(prompt, sequence_start=False) == prompt assert model.stop_words is None - assert model.repetition_penalty == 1.1 model = MODELS.get('baichuan-7b')(capability='chat') _prompt = model.get_prompt(prompt, sequence_start=True)