zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-29 06:54:39 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixes to config format in documentattion

Browse Source
This commit is contained in:
Chime Ogbuji 2024-11-03 20:00:35 -05:00 committed by Awni Hannun
parent 8ec802f468
commit 214c79be9c
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
llms/mlx_lm/LORA.md
View File

@ -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: each with the same structure as above. For example:
```yaml ```yaml
hf_datasets: [ hf_datasets:
"hf_dataset": - hf_dataset:
name: "Open-Orca/OpenOrca" name: "Open-Orca/OpenOrca"
train_split: "train[:90%]" train_split: "train[:90%]"
valid_split: "train[-10%:]" valid_split: "train[-10%:]"
prompt_feature: "question" prompt_feature: "question"
completion_feature: "response", completion_feature: "response"
"hf_dataset": - hf_dataset:
name: "trl-lib/ultrafeedback_binarized" name: "trl-lib/ultrafeedback_binarized"
train_split: "train[:90%]" train_split: "train[:90%]"
valid_split: "train[-10%:]" valid_split: "train[-10%:]"
chat_feature: "chosen" chat_feature: "chosen"
]
``` ```
- Arguments specified in `config` will be passed as keyword arguments to - Arguments specified in `config` will be passed as keyword arguments to