fix a bug in v1_embeeding_request (#2014)

This commit is contained in:
Xiaoyu Zhang
2024-11-12 16:49:45 +08:00
committed by GitHub
parent 78c1d6445f
commit a1bd719031

View File

@@ -1281,7 +1281,7 @@ def v1_embedding_request(all_requests, tokenizer_manager):
else:
prompt_kwargs = {"input_ids": prompt}
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}
else:
prompt_kwargs = {"input_ids": prompts}