From 6cc70cec473281d8dfcb2580e104c31428c03f18 Mon Sep 17 00:00:00 2001 From: Yu Meng Date: Sun, 2 Jun 2024 17:55:15 -0400 Subject: [PATCH] fix dataset --- training_configs/mistral-7b-instruct-simpo.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/training_configs/mistral-7b-instruct-simpo.yaml b/training_configs/mistral-7b-instruct-simpo.yaml index cc42119..8808423 100644 --- a/training_configs/mistral-7b-instruct-simpo.yaml +++ b/training_configs/mistral-7b-instruct-simpo.yaml @@ -5,7 +5,7 @@ use_flash_attention_2: true # Data training arguments dataset_mixer: - snorkelai/Snorkel-Mistral-PairRM-DPO-Dataset: 1.0 + princeton-nlp/mistral-instruct-ultrafeedback: 1.0 dataset_splits: - train - test