Merge pull request #34 from for-ai/gtranslate_updated

Added example command for translation
for-ai · Sep 9, 2024 · d230c18 · d230c18
2 parents 122131c + 83ebd79
commit d230c18
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scripts/translate_preference_pairs_gtranslate.py b/scripts/translate_preference_pairs_gtranslate.py
@@ -162,3 +162,5 @@ def translate_dataset(dataset, columns_to_translate, target_language, subset_siz
         args.output_dir,
         args.batch_size,
     )
+# Example Command:
+# python translate_preference_pairs_gtranslate.py --dataset_name allenai/reward-bench --target_language ar --columns_to_translate prompt chosen rejected --output_dir translated_results