add vllm config

This commit is contained in:
hiyouga
2024-11-10 21:28:18 +08:00
parent adc5849ce7
commit 58ab4579dc
34 changed files with 44 additions and 34 deletions

View File

@@ -12,7 +12,7 @@ pref_loss: sigmoid # choices: [sigmoid (dpo), orpo, simpo]
### dataset
dataset: dpo_en_demo
template: llama3
cutoff_len: 1024
cutoff_len: 2048
max_samples: 1000
overwrite_cache: true
preprocessing_num_workers: 16