From b8990d9078a2a88cfd956fd985fa443ac7342b6b Mon Sep 17 00:00:00 2001 From: theblackcat102 Date: Mon, 23 Jan 2023 02:48:47 +0000 Subject: [PATCH] [fix] remove spaces in format_pair --- model/supervised_finetuning/custom_datasets/formatting.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model/supervised_finetuning/custom_datasets/formatting.py b/model/supervised_finetuning/custom_datasets/formatting.py index 2f0adecd..a6c1c0d8 100644 --- a/model/supervised_finetuning/custom_datasets/formatting.py +++ b/model/supervised_finetuning/custom_datasets/formatting.py @@ -2,4 +2,4 @@ QA_SPECIAL_TOKENS = {"Question": "", "Answer": "", "StartPrefix": "< def format_pair(pair): - return "{} {} {}".format(QA_SPECIAL_TOKENS["Question"], pair[0], QA_SPECIAL_TOKENS["Answer"]), pair[1] + return "{}{}{}".format(QA_SPECIAL_TOKENS["Question"], pair[0], QA_SPECIAL_TOKENS["Answer"]), pair[1]