nodes/thecooltechguy-ComfyUI-Stable-Video-Diffusion
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

wip

Browse Source
This commit is contained in:
thecooltechguy
2023-11-23 23:40:36 +00:00
parent 02a17b8503
commit 3e8c12e52d
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+12
View File
@@ -303,6 +303,11 @@ class SVDSimpleImg2Vid:
checkpoint = os.path.join(folder_paths.get_folder_paths("svd")[0], checkpoint) 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 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( self.svd_model = load_model(
config=config, config=config,
device=device, device=device,
@@ -310,6 +315,13 @@ class SVDSimpleImg2Vid:
num_steps=num_steps, num_steps=num_steps,
checkpoint=checkpoint, 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 # convert image torch tensor to PIL image
# image shape: (1, H, W, C) # image shape: (1, H, W, C)