From f879ad4983855ca89e6c97a2ab0374d319f1cfcc Mon Sep 17 00:00:00 2001 From: ooooooye <53851298+brandneway@users.noreply.github.com> Date: Thu, 6 Aug 2026 11:56:55 +0800 Subject: [PATCH] fix(minimax-h3): pass device_type to fork_rng so VAE condition encode works on NPU devices (#5837) Signed-off-by: brandneway --- vllm_omni/diffusion/models/minimax_h3/vae.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vllm_omni/diffusion/models/minimax_h3/vae.py b/vllm_omni/diffusion/models/minimax_h3/vae.py index 5ee194e42..48cc5b8ca 100644 --- a/vllm_omni/diffusion/models/minimax_h3/vae.py +++ b/vllm_omni/diffusion/models/minimax_h3/vae.py @@ -193,7 +193,7 @@ class MiniMaxH3VideoVAE(nn.Module, DistributedVaeMixin): self.to(torch.float32) devices = [parameter.device] if parameter.device.type != "cpu" else [] try: - with torch.random.fork_rng(devices=devices): + with torch.random.fork_rng(devices=devices, device_type=parameter.device.type): torch.default_generator.manual_seed(MINIMAX_H3_KEYFRAME_ENCODE_SEED) for device in devices: with self.device_module.device(device): @@ -237,7 +237,7 @@ class MiniMaxH3VideoVAE(nn.Module, DistributedVaeMixin): self.to(torch.float32) devices = [parameter.device] if parameter.device.type != "cpu" else [] try: - with torch.random.fork_rng(devices=devices): + with torch.random.fork_rng(devices=devices, device_type=parameter.device.type): torch.default_generator.manual_seed(MINIMAX_H3_KEYFRAME_ENCODE_SEED) for device in devices: with self.device_module.device(device):