fix: don't pass around the same dict object (#1264)

pull/1270/head
Zach Nussbaum 1 year ago committed by GitHub
parent 41f640577c
commit b3f84c56e7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -117,14 +117,13 @@ async def completions(request: CompletionRequest):
params["num_return_sequences"] = request.n
header = {"Content-Type": "application/json"}
payload = {"parameters": params}
if isinstance(request.prompt, list):
tasks = []
for prompt in request.prompt:
payload = {"parameters": params}
payload["inputs"] = prompt
task = gpu_infer(payload, header)
tasks.append(task)
results = await asyncio.gather(*tasks)
choices = []
@ -147,6 +146,7 @@ async def completions(request: CompletionRequest):
)
else:
payload = {"parameters": params}
# If streaming, we need to return a StreamingResponse
payload["inputs"] = request.prompt

Loading…
Cancel
Save