From bd444232bb0dd4bcb95ae40bb269d2fdb8423c50 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 17 Feb 2024 15:20:44 -0600 Subject: [PATCH] fix parameters to get parameters --- api/onnx_web/server/api.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/server/api.py b/api/onnx_web/server/api.py index 0250e903..b30b3ac8 100644 --- a/api/onnx_web/server/api.py +++ b/api/onnx_web/server/api.py @@ -300,7 +300,7 @@ def img2img(server: ServerContext, pool: DevicePoolExecutor): def txt2img(server: ServerContext, pool: DevicePoolExecutor): - params = get_request_params() + params = get_request_params(server, JobType.TXT2IMG.value) replace_wildcards(params.image, get_wildcard_data()) job_name = make_job_name(JobType.TXT2IMG.value, params.image, params.size) @@ -406,7 +406,7 @@ def upscale(server: ServerContext, pool: DevicePoolExecutor): source = Image.open(BytesIO(source_file.read())).convert("RGB") - params = get_request_params(server) + params = get_request_params(server, JobType.UPSCALE.value) replace_wildcards(params.image, get_wildcard_data()) job_name = make_job_name("upscale", params.image, params.size)