Commit a36a30fb authored by Kohaku-Blueleaf's avatar Kohaku-Blueleaf
Browse files

add gc after using consistency dec

parent 2ea87265

modules/sd_vae_consistency.py

+1 −0
Original line number Diff line number Diff line
@@ -30,4 +30,5 @@ def decoder_model(): 
def unload():

    global sd_vae_consistency_models

    if sd_vae_consistency_models is not None:

        devices.torch_gc()

        sd_vae_consistency_models.ckpt.to('cpu')

CRA Git | Maintained and supported by SUSTech CRA and CCSE