From b0487fe92b936a0e402cc25f34910f20f846480e Mon Sep 17 00:00:00 2001 From: Sihan Chen <39623753+Spycsh@users.noreply.github.com> Date: Thu, 17 Oct 2024 15:48:44 +0800 Subject: [PATCH] fix chatqna accuracy issue with incorrect penalty (#974) --- ChatQnA/chatqna.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ChatQnA/chatqna.py b/ChatQnA/chatqna.py index 5fe6000f6..3ba1df7a9 100644 --- a/ChatQnA/chatqna.py +++ b/ChatQnA/chatqna.py @@ -67,8 +67,8 @@ def align_inputs(self, inputs, cur_node, runtime_graph, llm_parameters_dict, **k next_inputs["top_p"] = llm_parameters_dict["top_p"] next_inputs["stream"] = inputs["streaming"] next_inputs["frequency_penalty"] = inputs["frequency_penalty"] - next_inputs["presence_penalty"] = inputs["presence_penalty"] - next_inputs["repetition_penalty"] = inputs["repetition_penalty"] + # next_inputs["presence_penalty"] = inputs["presence_penalty"] + # next_inputs["repetition_penalty"] = inputs["repetition_penalty"] next_inputs["temperature"] = inputs["temperature"] inputs = next_inputs return inputs