Update reranker/rank_listwise_os_vllm.py
Browse files
reranker/rank_listwise_os_vllm.py
CHANGED
@@ -8,9 +8,8 @@ import torch
|
|
8 |
import numpy as np
|
9 |
from ftfy import fix_text
|
10 |
from vllm import LLM, SamplingParams, RequestOutput
|
11 |
-
from rankllm import Prompt
|
12 |
|
13 |
-
from .rankllm import PromptMode, RankLLM
|
14 |
from .result import Result
|
15 |
|
16 |
ALPH_START_IDX = ord('A') - 1
|
|
|
8 |
import numpy as np
|
9 |
from ftfy import fix_text
|
10 |
from vllm import LLM, SamplingParams, RequestOutput
|
|
|
11 |
|
12 |
+
from .rankllm import Prompt, PromptMode, RankLLM
|
13 |
from .result import Result
|
14 |
|
15 |
ALPH_START_IDX = ord('A') - 1
|