From b647acdb53d251cec126b79e505bac11821d7c93 Mon Sep 17 00:00:00 2001 From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> Date: Thu, 29 Feb 2024 04:49:01 +0100 Subject: [PATCH] FIX [`CI`] `require_read_token` in the llama FA2 test (#29361) Update test_modeling_llama.py --- tests/models/llama/test_modeling_llama.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/models/llama/test_modeling_llama.py b/tests/models/llama/test_modeling_llama.py index 308e5d9119..02c649c39a 100644 --- a/tests/models/llama/test_modeling_llama.py +++ b/tests/models/llama/test_modeling_llama.py @@ -398,6 +398,7 @@ class LlamaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi @require_torch_gpu @require_bitsandbytes @pytest.mark.flash_attn_test + @require_read_token @slow def test_flash_attn_2_generate_padding_right(self): """