From 214c79be9ceedde4ffb6e0838c720b8c7ab10107 Mon Sep 17 00:00:00 2001 From: Chime Ogbuji Date: Sun, 3 Nov 2024 20:00:35 -0500 Subject: [PATCH] Fixes to config format in documentattion --- llms/mlx_lm/LORA.md | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/llms/mlx_lm/LORA.md b/llms/mlx_lm/LORA.md index d553cecc..b8530708 100644 --- a/llms/mlx_lm/LORA.md +++ b/llms/mlx_lm/LORA.md @@ -304,19 +304,18 @@ You can specify a list of HF datasets using the `hf_datasets` (plural) configura each with the same structure as above. For example: ```yaml -hf_datasets: [ - "hf_dataset": +hf_datasets: +- hf_dataset: name: "Open-Orca/OpenOrca" train_split: "train[:90%]" valid_split: "train[-10%:]" prompt_feature: "question" - completion_feature: "response", - "hf_dataset": + completion_feature: "response" +- hf_dataset: name: "trl-lib/ultrafeedback_binarized" train_split: "train[:90%]" valid_split: "train[-10%:]" chat_feature: "chosen" -] ``` - Arguments specified in `config` will be passed as keyword arguments to