🔧 chore: relaese vmem before update pipeline

This commit is contained in:
Jianqi Pan
2024-05-08 20:12:10 +09:00
parent 55636490a3
commit aa0211ee28
+2
View File
@@ -452,6 +452,8 @@ class DiffusersPipeline:
scheduler_name: str = None,
use_tiny_vae: str = "disable",
):
torch.cuda.empty_cache()
gc.collect()
ckpt_path = folder_paths.get_full_path("checkpoints", ckpt_name)
if ckpt_path is None:
ckpt_path = ckpt_name