Skip to content

Commit 59a72cd

Browse files
fix: space missing from data_formatter_template causing mismatch with response_template (#455)
Signed-off-by: Harikrishnan Balagopal <[email protected]>
1 parent f32ac38 commit 59a72cd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ Example: Train.json
133133
},
134134
...
135135
]`
136-
data_formatter_template: `### Input: {{input}} \n\n##Label: {{output}}`
136+
data_formatter_template: `### Input: {{input}} \n\n## Label: {{output}}`
137137

138138
Formatting will happen on the fly while tuning. The keys in template should match fields in the dataset file. The `response template` corresponding to the above template will need to be supplied. in this case, `response template` = `\n## Label:`.
139139

@@ -299,7 +299,7 @@ python tuning/sft_trainer.py \
299299
--gradient_accumulation_steps 4 \
300300
--learning_rate 1e-5 \
301301
--response_template "\n## Label:" \
302-
--data_formatter_template: "### Input: {{input}} \n\n##Label: {{output}}"
302+
--data_formatter_template: "### Input: {{input}} \n\n## Label: {{output}}"
303303

304304
```
305305

0 commit comments

Comments
 (0)