fix dataset

This commit is contained in:
Yu Meng
2024-06-02 17:55:15 -04:00
parent 57312a6e3b
commit 6cc70cec47
@@ -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