Skip to content

Commit 2f605e0

Browse files
committed
Add WanVideoVACEEncode and enhance WanVideoEncode for multi-GPU support
- Implemented WanVideoVACEEncode class for encoding with VACE, including input parameters for width, height, number of frames, and strength. - Enhanced WanVideoEncode class to support multi-GPU encoding, with additional parameters for noise augmentation and latent strength. - Updated device management to ensure proper device context during encoding processes.
1 parent ba0a3a6 commit 2f605e0

3 files changed

Lines changed: 769 additions & 563 deletions

File tree

__init__.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,8 @@ def text_encoder_device_patched():
222222
WanVideoDecode,
223223
WanVideoModelLoader,
224224
WanVideoSampler,
225+
WanVideoVACEEncode,
226+
WanVideoEncode,
225227
)
226228

227229
from .wrappers import (
@@ -368,7 +370,9 @@ def register_and_count(module_names, node_map):
368370
"WanVideoImageToVideoEncodeMultiGPU": WanVideoImageToVideoEncode,
369371
"WanVideoDecodeMultiGPU": WanVideoDecode,
370372
"WanVideoModelLoaderMultiGPU": WanVideoModelLoader,
371-
"WanVideoSamplerMultiGPU": WanVideoSampler
373+
"WanVideoSamplerMultiGPU": WanVideoSampler,
374+
"WanVideoVACEEncodeMultiGPU": WanVideoVACEEncode,
375+
"WanVideoEncodeMultiGPU": WanVideoEncode,
372376
}
373377
register_and_count(["ComfyUI-WanVideoWrapper", "comfyui-wanvideowrapper"], wanvideo_nodes)
374378

0 commit comments

Comments
 (0)