Unverified Commit a292d2c4 authored by AngelBottomless's avatar AngelBottomless Committed by GitHub
Browse files

hotfix: call shared.state.end() after postprocessing done

parent 5e80d9ee

modules/postprocessing.py

+1 −1
Original line number Diff line number Diff line
@@ -78,7 +78,7 @@ def run_postprocessing(extras_mode, image, image_folder, input_dir, output_dir, 
        image_data.close()



    devices.torch_gc()



    shared.state.end()

    return outputs, ui_common.plaintext_to_html(infotext), ''

CRA Git | Maintained and supported by SUSTech CRA and CCSE