[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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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