[skip ci][Doc] Remove obsolete DiffusionParallelConfig in many example scripts (#6347)

Signed-off-by: Huang, Zeyu <11222265+fhfuih@users.noreply.github.com>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
This commit is contained in:
Zeyu Huang | 黃澤宇andHongsheng Liu authored and GitHub committed 2026-08-19 15:27:58 +08:00
1 parent 7099583404
commit 8dc3d504d8
10 files changed
+44 -92

No files matched your search

@@ -154,8 +154,6 @@ def _positive_finite(value: float, flag: str) -> float:
def build_omni_kwargs(
args: argparse.Namespace,
paths: LingBotPaths,
*,
parallel_config: Any,
) -> dict[str, Any]:
"""Build Omni configuration without overriding checkpoint class discovery."""
@@ -167,7 +165,7 @@ def build_omni_kwargs(
return {
"model": model,
"flow_shift": flow_shift,
"parallel_config": parallel_config,
"tensor_parallel_size": args.tensor_parallel_size,
"enforce_eager": args.enforce_eager,
"model_config": {"lingbot_action_root": str(paths.action_root)},
}
@@ -262,12 +260,10 @@ def main(argv: Sequence[str] | None = None) -> Path:
# helper tests work on machines without the native vLLM runtime.
from diffusers.utils import export_to_video
from vllm_omni.diffusion.data import DiffusionParallelConfig
from vllm_omni.entrypoints.omni import Omni
from vllm_omni.inputs.data import OmniDiffusionSamplingParams
parallel_config = DiffusionParallelConfig(tensor_parallel_size=args.tensor_parallel_size)
omni_kwargs = build_omni_kwargs(args, paths, parallel_config=parallel_config)
omni_kwargs = build_omni_kwargs(args, paths)
omni = Omni(**omni_kwargs)
try:
outputs = omni.generate(
@@ -125,7 +125,6 @@ async def run(argv: Sequence[str] | None = None) -> Path:
# require a CUDA-enabled vLLM installation.
import torch
from vllm_omni.diffusion.data import DiffusionParallelConfig
from vllm_omni.diffusion.models.lingbot_world.actions import LingBotCameraControlReducer
from vllm_omni.entrypoints.async_omni import AsyncOmni
from vllm_omni.experimental.ar_diffusion.consumer import ARDiffusionOmniTickConsumer
@@ -141,7 +140,7 @@ async def run(argv: Sequence[str] | None = None) -> Path:
model=args.model,
engine_backend="vllm_omni.experimental.ar_diffusion.engine.ARDiffusionEngine",
enforce_eager=args.enforce_eager,
parallel_config=DiffusionParallelConfig(tensor_parallel_size=args.tensor_parallel_size),
tensor_parallel_size=args.tensor_parallel_size,
max_num_seqs=1,
model_config={
"ar_diffusion_height": args.height,