From aa0211ee28e98eb54ac79f2bf9303edd1b133418 Mon Sep 17 00:00:00 2001
From: Jianqi Pan <jannchie@gmail.com>
Date: Wed, 8 May 2024 20:12:10 +0900
Subject: [PATCH] :wrench: chore: relaese vmem before update pipeline

---
 __init__.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/__init__.py b/__init__.py
index c3f8f29..4e0f0dd 100644
--- a/__init__.py
+++ b/__init__.py
@@ -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