mirror of
https://github.com/huggingface/trl.git
synced 2025-10-20 18:43:52 +08:00
Fix CI slow test OSError: You are trying to access a gated repo (#4283)
This commit is contained in:
committed by
GitHub
parent
927cf6ba46
commit
966b397201
@ -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")
|
||||
|
Reference in New Issue
Block a user