From 3e8c12e52d53557e1bf41fc689cd6b19bb6add38 Mon Sep 17 00:00:00 2001
From: thecooltechguy <subby@galiboo.com>
Date: Thu, 23 Nov 2023 23:40:36 +0000
Subject: [PATCH] wip

---
 nodes.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/nodes.py b/nodes.py
index fcc9fd3..5b54181 100644
--- a/nodes.py
+++ b/nodes.py
@@ -303,6 +303,11 @@ class SVDSimpleImg2Vid:
         checkpoint = os.path.join(folder_paths.get_folder_paths("svd")[0], checkpoint)
         
         if self.svd_model is None or self.is_model_config_different(config, device, num_frames, num_steps, checkpoint):
+            if self.svd_model:
+                del self.svd_model
+                gc.collect()
+                self.svd_model = None
+            
             self.svd_model = load_model(
                 config=config,
                 device=device,
@@ -310,6 +315,13 @@ class SVDSimpleImg2Vid:
                 num_steps=num_steps,
                 checkpoint=checkpoint,
             )
+            self.svd_config = config
+            self.device = device
+            self.num_frames = num_frames
+            self.num_steps = num_steps
+            self.checkpoint = checkpoint
+
+        model = self.svd_model
 
         # convert image torch tensor to PIL image
         # image shape: (1, H, W, C)