rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Remove reset switch for sd web api

Browse Source 
Automatically recover from errors
This commit is contained in:
somebody
2022-11-24 23:34:48 -06:00
parent a48d31c746
commit ec20b31931
3 changed files with 7 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
aiserver.py
View File

@@ -9384,7 +9384,11 @@ def text2img_api(prompt,
apiaddress = '{}/sdapi/v1/txt2img'.format(koboldai_vars.img_gen_api_url.rstrip("/"))
payload_json = json.dumps(final_imgen_params)
logger.debug(final_imgen_params)
submit_req = requests.post(url=apiaddress, data=payload_json)
try:
submit_req = requests.post(url=apiaddress, data=payload_json)
finally:
koboldai_vars.generating_image = False
if submit_req.status_code == 404:
show_error_notification(
@@ -9396,7 +9400,6 @@ def text2img_api(prompt,
elif not submit_req.ok:
show_error_notification("SD Web API Failure", f"HTTP Code {submit_req.status_code} -- See console for details")
logger.error(f"SD Web API Failure: HTTP Code {submit_req.status_code}, Body:\n{submit_req.text}")
koboldai_vars.generating_image = False
return None
results = submit_req.json()