diff options
author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-04-29 08:58:54 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-04-29 08:58:54 +0000 |
commit | f54cd3f1586ec8d20ffcaff184ba2e680bb8cb8b (patch) | |
tree | 7af57d3994e5db508c89f43858a67aefd917450b /modules/api/api.py | |
parent | 7fb72edaffd3d4f2336d2478a424fc455f2376a6 (diff) | |
parent | e55cb92067094564c259b515b8a6ec3869654083 (diff) | |
download | stable-diffusion-webui-gfx803-f54cd3f1586ec8d20ffcaff184ba2e680bb8cb8b.tar.gz stable-diffusion-webui-gfx803-f54cd3f1586ec8d20ffcaff184ba2e680bb8cb8b.tar.bz2 stable-diffusion-webui-gfx803-f54cd3f1586ec8d20ffcaff184ba2e680bb8cb8b.zip |
Merge branch 'dev' into torch
Diffstat (limited to 'modules/api/api.py')
-rw-r--r-- | modules/api/api.py | 12 |
1 files changed, 3 insertions, 9 deletions
diff --git a/modules/api/api.py b/modules/api/api.py index 518b2a61..5ed670e9 100644 --- a/modules/api/api.py +++ b/modules/api/api.py @@ -6,7 +6,6 @@ import uvicorn import gradio as gr from threading import Lock from io import BytesIO -from gradio.processing_utils import decode_base64_to_file from fastapi import APIRouter, Depends, FastAPI, Request, Response from fastapi.security import HTTPBasic, HTTPBasicCredentials from fastapi.exceptions import HTTPException @@ -395,16 +394,11 @@ class Api: def extras_batch_images_api(self, req: ExtrasBatchImagesRequest): reqDict = setUpscalers(req) - def prepareFiles(file): - file = decode_base64_to_file(file.data, file_path=file.name) - file.orig_name = file.name - return file - - reqDict['image_folder'] = list(map(prepareFiles, reqDict['imageList'])) - reqDict.pop('imageList') + image_list = reqDict.pop('imageList', []) + image_folder = [decode_base64_to_image(x.data) for x in image_list] with self.queue_lock: - result = postprocessing.run_extras(extras_mode=1, image="", input_dir="", output_dir="", save_output=False, **reqDict) + result = postprocessing.run_extras(extras_mode=1, image_folder=image_folder, image="", input_dir="", output_dir="", save_output=False, **reqDict) return ExtrasBatchImagesResponse(images=list(map(encode_pil_to_base64, result[0])), html_info=result[1]) |