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
|