diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-06-19 02:07:06 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-06-19 02:07:06 +0200 |
commit | aac821b148c6c0d35b940609dc9b0ddcb053b28e (patch) | |
tree | 5c125045b2b60ead39e093327d664adf43d1d35b /rag/retriever/rerank/local.py | |
parent | f2846429310452bebbf0d07203b1e53978c439c7 (diff) |
Still wip on rewrite
Diffstat (limited to 'rag/retriever/rerank/local.py')
-rw-r--r-- | rag/retriever/rerank/local.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/rag/retriever/rerank/local.py b/rag/retriever/rerank/local.py index e727165..fd42c2c 100644 --- a/rag/retriever/rerank/local.py +++ b/rag/retriever/rerank/local.py @@ -2,7 +2,7 @@ import os from typing import List from loguru import logger as log -from rag.rag import Message +from rag.message import Message from rag.retriever.vector import Document from sentence_transformers import CrossEncoder |