From 0d2044a7781b19677f22881e114c7b9159854bfa Mon Sep 17 00:00:00 2001 From: Yousef Rafat <81116377+yousef-rafat@users.noreply.github.com> Date: Fri, 19 Dec 2025 20:28:09 +0200 Subject: [PATCH] ... --- comfy_extras/nodes_seedvr.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/comfy_extras/nodes_seedvr.py b/comfy_extras/nodes_seedvr.py index e758ad516..bff358424 100644 --- a/comfy_extras/nodes_seedvr.py +++ b/comfy_extras/nodes_seedvr.py @@ -117,7 +117,6 @@ class SeedVR2InputProcessing(io.ComfyNode): @classmethod def execute(cls, images, vae, resolution_height, resolution_width): - comfy.model_management.load_models_gpu([vae.patcher]) device = vae.patcher.load_device offload_device = comfy.model_management.intermediate_device() @@ -145,7 +144,9 @@ class SeedVR2InputProcessing(io.ComfyNode): images = rearrange(images, "b t c h w -> b c t h w") images = images.to(device) + vae_model = vae_model.to(device) latent = vae_model.encode(images)[0] + vae_model = vae_model.to(offload_device) latent = latent.unsqueeze(2) if latent.ndim == 4 else latent latent = rearrange(latent, "b c ... -> b ... c")