From 736f46fb00f4f00a5c349d83ff329e747f8cbb2d Mon Sep 17 00:00:00 2001 From: theblackcat102 Date: Sun, 22 Jan 2023 14:00:20 +0000 Subject: [PATCH] [fix] prosocial dialogue format error --- .../supervised_finetuning/custom_datasets/toxic_conversation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model/supervised_finetuning/custom_datasets/toxic_conversation.py b/model/supervised_finetuning/custom_datasets/toxic_conversation.py index d3433106..640b8d8d 100644 --- a/model/supervised_finetuning/custom_datasets/toxic_conversation.py +++ b/model/supervised_finetuning/custom_datasets/toxic_conversation.py @@ -67,4 +67,4 @@ class ProsocialDialogue(Dataset): return len(self.pairs) def __getitem__(self, idx): - return format_pair(self.pairs[idx]) + return self.pairs[idx]