Comments

bigscience-workshop · Apr 10, 2024 · 5ca2a03 · 5ca2a03
1 parent 63a421b
commit 5ca2a03
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_full_model.py b/tests/test_full_model.py
@@ -142,8 +142,8 @@ def test_sampling(tokenizer, model, ref_model, max_new_tokens=10):
 
 
 @pytest.mark.skipif(
- MODEL_NAME.lower().find("bloom") == -1,
- reason="Mixtral and Llama uses DynamicCache, which can change based on beam search choices",
+ "bloom" not in MODEL_NAME.lower(),
+ reason="Mixtral and Llama use 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):