[CI/Build] Fix CPU groups in distributed VAE test mocks (#8591)

This commit is contained in:
amy-why-3459 authored and GitHub committed 2026-10-07 10:24:15 -06:00
1 parent 32ead8e5c8
commit 3345fb7498
2 files changed
+3 -2

No files matched your search

@@ -89,7 +89,7 @@ def _patch_parallel_worker(rank: int, port: str, shapes: dict) -> None:
try:
import vllm_omni.diffusion.distributed.autoencoders.distributed_vae_executor as executor_mod
world = SimpleNamespace(device_group=dist.group.WORLD)
world = SimpleNamespace(device_group=dist.group.WORLD, cpu_group=dist.group.WORLD)
executor_mod.get_world_group = lambda: world
vae = _stub_vae()