riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-03 21:00:08 +06:00
Code Issues Actions 25 Packages Projects Releases Wiki Activity

Update CsmForConditionalGenerationIntegrationTest (#38424)

Browse Source 
* require_read_token

* ruff

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2025-05-28 10:20:43 +02:00 committed by GitHub
parent baddbdd24b
commit cea254c909
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/csm/test_modeling_csm.py
View File

@ -30,6 +30,7 @@ from transformers import (
) )
from transformers.testing_utils import ( from transformers.testing_utils import (
cleanup, cleanup,
require_read_token,
require_torch_accelerator, require_torch_accelerator,
slow, slow,
torch_device, torch_device,
@ -414,6 +415,7 @@ class CsmForConditionalGenerationTest(ModelTesterMixin, GenerationTesterMixin, u
return input_ids, position_ids, input_ids_shared_prefix, mask_shared_prefix, position_ids_shared_prefix return input_ids, position_ids, input_ids_shared_prefix, mask_shared_prefix, position_ids_shared_prefix
@require_read_token
class CsmForConditionalGenerationIntegrationTest(unittest.TestCase): class CsmForConditionalGenerationIntegrationTest(unittest.TestCase):
def setUp(self): def setUp(self):
# TODO: @eustlb, update with correct sesame's repo # TODO: @eustlb, update with correct sesame's repo