From 6a1596b63340d71b6cccd12980882d041cc5dd1a Mon Sep 17 00:00:00 2001 From: Artem Chumachenko Date: Wed, 10 Apr 2024 10:32:04 +0200 Subject: [PATCH] Update tests/test_full_model.py Co-authored-by: Max Ryabinin --- tests/test_full_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_full_model.py b/tests/test_full_model.py index 73e8be0..30a9bed 100644 --- a/tests/test_full_model.py +++ b/tests/test_full_model.py @@ -142,7 +142,7 @@ def test_sampling(tokenizer, model, ref_model, max_new_tokens=10): @pytest.mark.skipif( - MODEL_NAME.lower().find("mixtral"), reason="Mixtral use DynamicCache, that can change based on BS choices" + MODEL_NAME.lower().find("mixtral"), reason="Mixtral uses DynamicCache, which can change based on beam search choices" ) @pytest.mark.forked def test_beam_search_generation(tokenizer, model, ref_model, max_new_tokens=4, num_beams=5):