diff --git a/examples/trl/dpo.py b/examples/trl/dpo.py index 9804a6f04a..f69081123e 100644 --- a/examples/trl/dpo.py +++ b/examples/trl/dpo.py @@ -114,7 +114,6 @@ def get_stack_exchange_paired( split="train", cache_dir=cache_dir, data_dir=data_dir, - verification_mode="no_checks", # TODO: remove this line when Datasets v2.19.2 is released ) original_columns = dataset.column_names diff --git a/examples/trl/requirements.txt b/examples/trl/requirements.txt index 200cfa1b9f..01f0e51a80 100644 --- a/examples/trl/requirements.txt +++ b/examples/trl/requirements.txt @@ -1,6 +1,6 @@ trl == 0.8.6 peft == 0.6.2 -datasets +datasets == 2.19.2 tyro evaluate scikit-learn