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)