diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-12-17 05:55:35 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-12-17 05:55:35 +0000 |
| commit | de03882d6ca56bc81058f5120f028678a6a54aaa (patch) | |
| tree | 66df39ff95561dc0503eba5468d36b3ed10d55b0 /modules/api/api.py | |
| parent | 3d9a0d9e4b1a52a873d32c7ba5c462d840f0e3ee (diff) | |
| download | stable-diffusion-webui-gfx803-de03882d6ca56bc81058f5120f028678a6a54aaa.tar.gz stable-diffusion-webui-gfx803-de03882d6ca56bc81058f5120f028678a6a54aaa.tar.bz2 stable-diffusion-webui-gfx803-de03882d6ca56bc81058f5120f028678a6a54aaa.zip | |
make task ids for API work without force_task_id
Diffstat (limited to 'modules/api/api.py')
| -rw-r--r-- | modules/api/api.py | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/modules/api/api.py b/modules/api/api.py index 9637cb81..7154c9d5 100644 --- a/modules/api/api.py +++ b/modules/api/api.py @@ -336,9 +336,8 @@ class Api: return script_args def text2imgapi(self, txt2imgreq: models.StableDiffusionTxt2ImgProcessingAPI): - task_id = create_task_id("text2img") - if txt2imgreq.force_task_id is None: - task_id = txt2imgreq.force_task_id + task_id = txt2imgreq.force_task_id or create_task_id("txt2img") + script_runner = scripts.scripts_txt2img if not script_runner.scripts: script_runner.initialize_scripts(False) @@ -393,9 +392,7 @@ class Api: return models.TextToImageResponse(images=b64images, parameters=vars(txt2imgreq), info=processed.js()) def img2imgapi(self, img2imgreq: models.StableDiffusionImg2ImgProcessingAPI): - task_id = create_task_id("img2img") - if img2imgreq.force_task_id is None: - task_id = img2imgreq.force_task_id + task_id = img2imgreq.force_task_id or create_task_id("img2img") init_images = img2imgreq.init_images if init_images is None: |