提交 a36a30fb 编写于 作者: K Kohaku-Blueleaf

add gc after using consistency dec

上级 2ea87265
......@@ -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')
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册