[bugfix] make args.stream work (#6831)

This commit is contained in:
Wang Ran (汪然)
2024-07-27 17:07:02 +08:00
committed by GitHub
parent 925de97e05
commit a57d75821c

View File

@ -31,7 +31,10 @@ def post_http_request(prompt: str,
"max_tokens": 16,
"stream": stream,
}
response = requests.post(api_url, headers=headers, json=pload, stream=True)
response = requests.post(api_url,
headers=headers,
json=pload,
stream=stream)
return response