Update CsmForConditionalGenerationIntegrationTest (#38424)

* 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

View File

@@ -30,6 +30,7 @@ from transformers import (
)
from transformers.testing_utils import (
cleanup,
require_read_token,
require_torch_accelerator,
slow,
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
@require_read_token
class CsmForConditionalGenerationIntegrationTest(unittest.TestCase):
def setUp(self):
# TODO: @eustlb, update with correct sesame's repo