diff --git a/tests/test_sft_trainer.py b/tests/test_sft_trainer.py index 924010c6f4d..25e3c7ccd2b 100644 --- a/tests/test_sft_trainer.py +++ b/tests/test_sft_trainer.py @@ -321,6 +321,7 @@ def test_init_auto_processing_class_uses_model_revision(self): ), "trl-internal-testing/tiny-GptOssForCausalLM", "trl-internal-testing/tiny-Qwen2ForCausalLM-2.5", + "trl-internal-testing/tiny-Qwen2ForCausalLM-R1-Distill", "trl-internal-testing/tiny-Qwen3MoeForCausalLM", pytest.param( "trl-internal-testing/tiny-NemotronHForCausalLM-nano",