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

Fix for image generation using horde

Browse Source
This commit is contained in:
ebolam
2022-11-09 19:36:03 -05:00
parent 87e3f75ebe
commit 0cd70fdb56
1 changed files with 20 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
aiserver.py
View File

@@ -9223,24 +9223,33 @@ def text2img_horde(prompt,
filename = "story_art.png"): filename = "story_art.png"):
logger.debug("Generating Image using Horde") logger.debug("Generating Image using Horde")
koboldai_vars.generating_image = True koboldai_vars.generating_image = True
final_imgen_params = {
"n": 1,
"width": 512,
"height": 512,
"steps": 50,
}
final_submit_dict = { final_submit_dict = {
"prompt": "{}, {}".format(prompt, art_guide), "prompt": "{}, {}".format(prompt, art_guide),
"api_key": koboldai_vars.sh_apikey if koboldai_vars.sh_apikey != '' else "0000000000", "trusted_workers": False,
"params": final_imgen_params, "models": [
"stable_diffusion"
],
"params": {
"n":1,
"nsfw": True,
"sampler_name": "k_euler_a",
"karras": True,
"cfg_scale": 7.0,
"steps":25,
"width":512,
"height":512}
} }
cluster_headers = {'apikey': koboldai_vars.sh_apikey if koboldai_vars.sh_apikey != '' else "0000000000",}
logger.debug(final_submit_dict) logger.debug(final_submit_dict)
submit_req = requests.post('https://stablehorde.net/api/v1/generate/sync', json = final_submit_dict) submit_req = requests.post('https://stablehorde.net/api/v2/generate/sync', json = final_submit_dict, headers=cluster_headers)
if submit_req.ok: if submit_req.ok:
results = submit_req.json() results = submit_req.json()
for iter in range(len(results)): for iter in range(len(results['generations'])):
b64img = results[iter]["img"] b64img = results['generations'][iter]["img"]
base64_bytes = b64img.encode('utf-8') base64_bytes = b64img.encode('utf-8')
img_bytes = base64.b64decode(base64_bytes) img_bytes = base64.b64decode(base64_bytes)
img = Image.open(BytesIO(img_bytes)) img = Image.open(BytesIO(img_bytes))