Unverified Commit a1bd7190 authored by Xiaoyu Zhang's avatar Xiaoyu Zhang Committed by GitHub
Browse files

fix a bug in v1_embeeding_request (#2014)

parent 78c1d644
...@@ -1281,7 +1281,7 @@ def v1_embedding_request(all_requests, tokenizer_manager): ...@@ -1281,7 +1281,7 @@ def v1_embedding_request(all_requests, tokenizer_manager):
else: else:
prompt_kwargs = {"input_ids": prompt} prompt_kwargs = {"input_ids": prompt}
else: else:
if isinstance(prompts[0], str) or isinstance(propmt[0][0], str): if isinstance(prompts[0], str) or isinstance(propmts[0][0], str):
prompt_kwargs = {"text": prompts} prompt_kwargs = {"text": prompts}
else: else:
prompt_kwargs = {"input_ids": prompts} prompt_kwargs = {"input_ids": prompts}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment