Skip to content

S
Stable Diffusion Webui
Commit 6cc5a886 authored by caoxipeng's avatar caoxipeng
Browse files
Options

Add total_tqdm clear in the end of txt2img & img2img api.

parent 68f336bd
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
modules/api/api.py
View file @ 6cc5a886
...@@ -343,6 +343,7 @@ class Api: ...@@ -343,6 +343,7 @@ class Api:
processed = process_images(p) processed = process_images(p)
finally: finally:
shared.state.end() shared.state.end()
shared.total_tqdm.clear()
b64images = list(map(encode_pil_to_base64, processed.images)) if send_images else [] b64images = list(map(encode_pil_to_base64, processed.images)) if send_images else []
...@@ -402,6 +403,7 @@ class Api: ...@@ -402,6 +403,7 @@ class Api:
processed = process_images(p) processed = process_images(p)
finally: finally:
shared.state.end() shared.state.end()
shared.total_tqdm.clear()
b64images = list(map(encode_pil_to_base64, processed.images)) if send_images else [] b64images = list(map(encode_pil_to_base64, processed.images)) if send_images else []
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment