diff options
| author | gayshub <zhangxianglei1234@gmail.com> | 2023-12-15 16:57:17 +0800 |
|---|---|---|
| committer | gayshub <zhangxianglei1234@gmail.com> | 2023-12-15 16:57:17 +0800 |
| commit | 1242ba08e19f3d317bdc5924db2b73d0c9569a7f (patch) | |
| tree | a64af2bce85b280f8bb29fb91f878e6d19df9cf3 /modules/api/models.py | |
| parent | 4afaaf8a020c1df457bcf7250cb1c7f609699fa7 (diff) | |
add allow specify the task id and get the location of task in the queue of pending task
Diffstat (limited to 'modules/api/models.py')
| -rw-r--r-- | modules/api/models.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/api/models.py b/modules/api/models.py index 6a574771..7b7f1773 100644 --- a/modules/api/models.py +++ b/modules/api/models.py @@ -109,6 +109,7 @@ StableDiffusionTxt2ImgProcessingAPI = PydanticModelGenerator( {"key": "send_images", "type": bool, "default": True}, {"key": "save_images", "type": bool, "default": False}, {"key": "alwayson_scripts", "type": dict, "default": {}}, + {"key": "force_task_id", "type": str, "default": None}, ] ).generate_model() @@ -126,6 +127,7 @@ StableDiffusionImg2ImgProcessingAPI = PydanticModelGenerator( {"key": "send_images", "type": bool, "default": True}, {"key": "save_images", "type": bool, "default": False}, {"key": "alwayson_scripts", "type": dict, "default": {}}, + {"key": "force_task_id", "type": str, "default": None}, ] ).generate_model() |