SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

FIX [CI] require_read_token in the llama FA2 test (#29361)

Browse Source 
Update test_modeling_llama.py
This commit is contained in:
Younes Belkada
2024-02-29 04:49:01 +01:00
committed by GitHub
parent 8d8ac9c2df
commit b647acdb53
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/models/llama/test_modeling_llama.py
View File

@@ -398,6 +398,7 @@ class LlamaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
@require_torch_gpu @require_torch_gpu
@require_bitsandbytes @require_bitsandbytes
@pytest.mark.flash_attn_test @pytest.mark.flash_attn_test
@require_read_token
@slow @slow
def test_flash_attn_2_generate_padding_right(self): def test_flash_attn_2_generate_padding_right(self):
""" """