From 140f5ad4c3b582bb1b752efdd12b7b5e27fcca38 Mon Sep 17 00:00:00 2001 From: Russell Richardson Date: Wed, 12 Mar 2025 08:16:23 -0400 Subject: [PATCH] Allow passing firstpass_image to txt2img API This change allows passing in a base64 encoded image string to the `/sdapi/v1/txt2img` endpoint via the `firstpass_image` parameter. Can be useful when applications utilizing the API generate an initial image and then need that image upscaled using HiRes Fix - by replaying the request alongside the `firstpass_image` parameter with the base64 image returned by the original request, it bypasses the need to regenerate the original image since it has already been passed in. At least, that seems to be the original intent of this parameter (and works great with this change). --- modules/api/api.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modules/api/api.py b/modules/api/api.py index db2fc6480..0bc2e3e92 100644 --- a/modules/api/api.py +++ b/modules/api/api.py @@ -470,6 +470,8 @@ class Api: p.scripts = script_runner p.outpath_grids = opts.outdir_txt2img_grids p.outpath_samples = opts.outdir_txt2img_samples + if p.firstpass_image is not None: + p.firstpass_image = decode_base64_to_image(p.firstpass_image) try: shared.state.begin(job="scripts_txt2img")