Fix CI slow test OSError: You are trying to access a gated repo (#4283)

This commit is contained in:
Albert Villanova del Moral
2025-10-15 16:11:11 +02:00
committed by GitHub
parent 927cf6ba46
commit 966b397201

View File

@ -1477,6 +1477,7 @@ class TestSFTTrainer(TrlTestCase):
# To ensure coverage, we run tests on the full model but mark them as slow to exclude from default runs.
@pytest.mark.slow
@require_vision
@pytest.mark.skip(reason="Model google/gemma-3n-E2B-it is gated and requires HF token")
def test_train_vlm_gemma_3n(self):
# Get the dataset
dataset = load_dataset("trl-internal-testing/zen-image", "conversational_language_modeling", split="train")