wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-07 12:50:49 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

...

Browse Source
This commit is contained in:
Yousef Rafat 2025-12-19 20:28:09 +02:00
parent 7e62f8cc9f
commit 0d2044a778
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy_extras/nodes_seedvr.py
View File

@ -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")