Add no_checks for sub dataset in lvwerra/stack-exchange-paired since … (

#1003) Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
huggingface · Jun 5, 2024 · 1911dc6 · 1911dc6
1 parent aa403ee
commit 1911dc6
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git 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
@@ -1,6 +1,6 @@
 trl == 0.8.6
 peft == 0.6.2
-datasets
+datasets == 2.19.2
 tyro
 evaluate
 scikit-learn