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

Add basic authentication option for webUI image generation

Browse Source
This commit is contained in:
ebolam
2023-02-09 10:54:38 -05:00
parent 5aca142034
commit b29d4cef29
5 changed files with 40 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
aiserver.py
View File

@@ -9932,7 +9932,12 @@ def text2img_api(prompt, art_guide="") -> Image.Image:
logger.debug(final_imgen_params)
try:
submit_req = requests.post(url=apiaddress, data=payload_json)
logger.info("Gen Image API: Username: {}".format(koboldai_vars.img_gen_api_username))
if koboldai_vars.img_gen_api_username != "":
basic = requests.auth.HTTPBasicAuth(koboldai_vars.img_gen_api_username, koboldai_vars.img_gen_api_password)
submit_req = requests.post(url=apiaddress, data=payload_json, auth=basic)
else:
submit_req = requests.post(url=apiaddress, data=payload_json)
except requests.exceptions.ConnectionError:
show_error_notification(
"SD Web API Failure",