From cea254c909d956f3c4d09e297bb8802da5c05bfd Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 28 May 2025 10:20:43 +0200 Subject: [PATCH] Update `CsmForConditionalGenerationIntegrationTest` (#38424) * require_read_token * ruff --------- Co-authored-by: ydshieh --- tests/models/csm/test_modeling_csm.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/models/csm/test_modeling_csm.py b/tests/models/csm/test_modeling_csm.py index 2425ff35ecf..15467f5e1ba 100644 --- a/tests/models/csm/test_modeling_csm.py +++ b/tests/models/csm/test_modeling_csm.py @@ -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