update training args check for new defaults (#3051) [skip ci]

* update training args check for new defaults

* skip check for now
This commit is contained in:
Wing Lian
2025-08-10 11:26:22 -04:00
committed by GitHub
parent 05f1b4b2e8
commit d6b81b3683

View File

@@ -281,7 +281,9 @@ class TestHFRLTrainerBuilder:
# Other settings
assert training_arguments.dataloader_num_workers == 1
assert training_arguments.dataloader_pin_memory is True
assert training_arguments.gradient_checkpointing is False
# TODO(wing): restore once trl releases 0.22.0
# assert training_arguments.gradient_checkpointing is True
def test_dpo_training_arguments(self, dpo_cfg, model, tokenizer):
builder = HFRLTrainerBuilder(dpo_cfg, model, tokenizer)