Commit 6cc5a886 authored by caoxipeng's avatar caoxipeng
Browse files

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

parent 68f336bd

modules/api/api.py

+2 −0
Original line number Diff line number Diff line
@@ -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 []

CRA Git | Maintained and supported by SUSTech CRA and CCSE