Skip to content

Commit 5b4f31e

Browse files
committed
fix vl_rerank
1 parent 6eccefc commit 5b4f31e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

gpt_server/model_worker/embedding.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,8 @@ async def get_embeddings(self, params):
112112
query_type=query_type,
113113
doc_type=doc_type,
114114
)
115+
if isinstance(scores, float):
116+
scores = [scores]
115117
embedding = [[float(score)] for score in scores]
116118
elif self.mode == "clip_text_model":
117119
token_num = 0

0 commit comments

Comments
 (0)