Fix dataset in GKD script (#2067)

I added the wrong dataset name in a prior commit 🙈
huggingface · Sep 14, 2024 · 08ba866 · 08ba866
1 parent ebc85b2
commit 08ba866
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/scripts/gkd.py b/examples/scripts/gkd.py
@@ -17,7 +17,7 @@
 python examples/scripts/gkd.py \
     --model_name_or_path Qwen/Qwen2-0.5B-Instruct \
     --teacher_model_name_or_path Qwen/Qwen2-1.5B-Instruct \
-    --dataset_name trl-lib/chatbot_arena_conversations \
+    --dataset_name trl-lib/chatbot_arena_completions \
     --learning_rate 2e-5 \
     --per_device_train_batch_size 4 \
     --gradient_accumulation_steps 8 \
@@ -31,7 +31,7 @@
 python examples/scripts/gkd.py \
     --model_name_or_path Qwen/Qwen2-0.5B-Instruct \
     --teacher_model_name_or_path Qwen/Qwen2-1.5B-Instruct \
-    --dataset_name trl-lib/chatbot_arena_conversations \
+    --dataset_name trl-lib/chatbot_arena_completions \
     --learning_rate 2e-4 \
     --per_device_train_batch_size 4 \
     --gradient_accumulation_steps 8 \