".github/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "48e36353d8cbf0322ec1ad0684b95d11f70af2de"
Commit a6aa46dd authored by Ying Sheng's avatar Ying Sheng
Browse files

minor

parent 405f26b0
...@@ -91,6 +91,8 @@ class HttpResponse: ...@@ -91,6 +91,8 @@ class HttpResponse:
def http_request(url, json=None, stream=False, auth_token=None): def http_request(url, json=None, stream=False, auth_token=None):
"""A faster version of requests.post with low-level urllib API.""" """A faster version of requests.post with low-level urllib API."""
if stream: if stream:
if auth_token is None:
return requests.post(url, json=json, stream=True)
headers = { headers = {
"Content-Type": "application/json", "Content-Type": "application/json",
"Authentication": f"Bearer {auth_token}" "Authentication": f"Bearer {auth_token}"
...@@ -99,7 +101,8 @@ def http_request(url, json=None, stream=False, auth_token=None): ...@@ -99,7 +101,8 @@ def http_request(url, json=None, stream=False, auth_token=None):
else: else:
req = urllib.request.Request(url) req = urllib.request.Request(url)
req.add_header("Content-Type", "application/json; charset=utf-8") req.add_header("Content-Type", "application/json; charset=utf-8")
req.add_header("Authentication", f"Bearer {auth_token}") if auth_token is not None:
req.add_header("Authentication", f"Bearer {auth_token}")
if json is None: if json is None:
data = None data = None
else: else:
......
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