From ead514a4f44e211c9cd735f089c0d06a733c2e75 Mon Sep 17 00:00:00 2001
From: Fanyou Wu <fanyou.wu@outlook.com>
Date: Sun, 16 Jun 2024 21:05:42 -0400
Subject: [PATCH] Update
---
models/v4.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/models/v4.py b/models/v4.py
index 5a4dc7c..104bf61 100644
--- a/models/v4.py
+++ b/models/v4.py
@@ -160,14 +160,12 @@ class V4(Base):
search_results = self.remove_duplicated_and_sort_trusted(
batch["search_results"]
)
- # search_results = self.remove_duplicated_and_sort_trusted(
- # batch["search_results"], domains=domains
- # )
search_results = [x[: self.config.max_num_doc] for x in search_results]
contents = self.content_retriever(
queries, search_results, final_answers_flag
)
+
query_times = batch["query_time"]
answers = self.qa_with_ref(
queries,
--
GitLab