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

This commit is contained in:
caoxipeng 2023-07-28 11:40:10 +08:00
parent 68f336bd99
commit 6cc5a886ae
1 changed files with 2 additions and 0 deletions

View File

@ -343,6 +343,7 @@ class Api:
processed = process_images(p)
finally:
shared.state.end()
shared.total_tqdm.clear()
b64images = list(map(encode_pil_to_base64, processed.images)) if send_images else []
@ -402,6 +403,7 @@ class Api:
processed = process_images(p)
finally:
shared.state.end()
shared.total_tqdm.clear()
b64images = list(map(encode_pil_to_base64, processed.images)) if send_images else []