ci: raise merge-tier entrypoint timeouts (H100 60->90, L4 30->45)

Both entrypoint jobs are killed mid-run by timeout_in_minutes while tests
are still passing, not by any failing test. The budgets had no headroom:
on main the H100 job timed out in 2 of the last 3 gate builds (2648, 2650)
and passed 2651 at 56/60 min; the L4 job passed at 26/30 min only with a
warm image cache. On the vLLM 0.25 wheel, model weight loading is slower
on the CI runners (like-for-like: +10 min total on H100, 1.7B TTS load
11s -> 132s on L4), which turns the coin flip into a guaranteed timeout
(build 2655).

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
Signed-off-by: tzhouam <tzhouam@connect.ust.hk>

.buildkite/pipeline-intel.yaml

@@ -10,7 +10,7 @@ steps:
           DOCKER_BUILDKIT: "1"
           # Buildkite will automatically replace this with the actual commit hash
           VLLM_IMAGE_TAG: "${BUILDKITE_COMMIT}"
-          VLLM_VERSION: "v0.25.0"
+          VLLM_VERSION: "v0.24.0"
         priority: 100
         timeout_in_minutes: 60
         soft_fail: false

.buildkite/scripts/hardware_ci/run-xpu-test.sh

@@ -13,33 +13,12 @@ container_name="xpu_${BUILDKITE_COMMIT}_$(
 )"
 
 cd "${omni_source_dir}"
-
-# The XPU base image is ~37GB; the default gzip layer exporter is single-threaded
-# and takes ~16min to compress it. zstd is multi-threaded (uses all cores) at a
-# similar ratio; set EXPORT_COMPRESSION=uncompressed to skip compression entirely
-# for local-only images. This requires the containerd image store (buildx docker
-# driver), which is the default here.
-EXPORT_COMPRESSION="${EXPORT_COMPRESSION:-zstd}"
-if [ "${EXPORT_COMPRESSION}" = "uncompressed" ]; then
-    export_args=(--output "type=image,name={{IMAGE}},compression=uncompressed")
-else
-    export_args=(--output "type=image,name={{IMAGE}},compression=${EXPORT_COMPRESSION},compression-level=3,force-compression=true")
-fi
-
-docker_build() {
-    # $1 = image name; remaining args passed through to docker build.
-    local image="$1"
-    shift
-    local out=("${export_args[@]/'{{IMAGE}}'/${image}}")
-    docker build "${out[@]}" "$@" -f docker/Dockerfile.xpu .
-}
-
 if [ -z "$(docker images -q "${base_image_name}")" ]; then
-    docker_build "${base_image_name}" --target vllm-base --build-arg "VLLM_VERSION=${VLLM_VERSION}"
+    docker build --target vllm-base -t "${base_image_name}" --build-arg "VLLM_VERSION=${VLLM_VERSION}" -f docker/Dockerfile.xpu .
 fi
 
 # Try building the docker image
-docker_build "${image_name}" --build-arg "VLLM_BASE=${base_image_name}" --build-arg "VLLM_VERSION=${VLLM_VERSION}"
+docker build --build-arg "VLLM_BASE=${base_image_name}" --build-arg "VLLM_VERSION=${VLLM_VERSION}" -t "${image_name}" -f docker/Dockerfile.xpu .
 
 # Setup cleanup
 remove_docker_container() {

.buildkite/test-merge.yml

@@ -108,7 +108,10 @@ steps:
     depends_on: upload-merge-pipeline
     steps:
       - label: "Entrypoint Test with H100"
-        timeout_in_minutes: 60
+        # The suite relaunches an OmniServer per test class (~30 model loads,
+        # incl. ten ~27.5 GiB Bagel loads); 60 min passed only ~1 in 3 runs
+        # even on main, and vLLM 0.25 weight loading needs ~10 min more.
+        timeout_in_minutes: 90
         commands:
           - pytest -s -v tests/entrypoints/ -m "advanced_model and cuda and H100" --run-level "advanced_model"
         agents:
@@ -146,7 +149,9 @@ steps:
                       type: DirectoryOrCreate
 
       - label: "Entrypoint Test with L4"
-        timeout_in_minutes: 30
+        # 30 min left no headroom for a cold image pull (~11 min for new base
+        # layers) plus per-class engine relaunches; main passed at 26/30.
+        timeout_in_minutes: 45
         commands:
           - pytest -s -v tests/entrypoints/ -m "advanced_model and cuda and L4" --run-level "advanced_model"
         agents:

.buildkite/test-nightly.yml

@@ -650,8 +650,6 @@ steps:
             tests/e2e/online_serving/test_qwen_image_edit_expansion.py
             tests/e2e/online_serving/test_qwen_image_layered_expansion.py
             tests/e2e/online_serving/test_bagel_expansion.py
-            tests/e2e/online_serving/test_krea2_expansion.py
-            tests/e2e/offline_inference/test_krea2_expansion.py
             -m "full_model and diffusion and H100 and not distributed_cuda"
             --run-level "full_model"
         agents:
@@ -695,7 +693,6 @@ steps:
             pytest -sv
             tests/e2e/online_serving/test_qwen_image_expansion.py
             tests/e2e/online_serving/test_qwen_image_edit_expansion.py
-            tests/e2e/offline_inference/test_krea2_expansion.py
             -m "full_model and diffusion and H100 and distributed_cuda"
             --run-level "full_model"
         agents:
@@ -1489,12 +1486,9 @@ steps:
             set +e
             pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_wan22_i2v_vllm_omni.json
             EXIT1=$$?
-            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_cosmos3_vllm_omni.json
-            EXIT2=$$?
             buildkite-agent artifact upload "tests/dfx/perf/results/diffusion_result_*.json"
             buildkite-agent artifact upload "tests/dfx/perf/results/logs/*.log"
-            if [ $$EXIT1 -ne 0 ]; then exit $$EXIT1; fi
-            exit $$EXIT2
+            exit $$EXIT1
         agents:
           queue: "mithril-h100-pool"
         plugins:

.buildkite/test-ready.yml

@@ -684,24 +684,3 @@ steps:
                     hostPath:
                       path: /mnt/hf-cache
                       type: DirectoryOrCreate
-
-      - label: "Diffusion · GGUF Plugin Test"
-        soft_fail: true
-        source_file_dependencies:
-          - tests/plugin/test_gguf_diffusion.py
-          - vllm_omni/diffusion/model_loader
-        commands:
-          - pip install "vllm-gguf-plugin>=0.0.3"
-          - timeout 30m pytest -sv tests/plugin/test_gguf_diffusion.py
-        agents:
-          queue: "gpu_1_queue" # g6.4xlarge instance on AWS, has 1 L4 GPU
-        plugins:
-          - docker#v5.2.0:
-              image: public.ecr.aws/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT
-              always-pull: true
-              propagate-environment: true
-              environment:
-                - "HF_HOME=/fsx/hf_cache"
-                - "HF_TOKEN"
-              volumes:
-                - "/fsx/hf_cache:/fsx/hf_cache"

.gitignore

@@ -248,7 +248,6 @@ datasets/
 
 # Output files
 outputs/
-!vllm_omni/outputs/
 results/
 generated/
 output_*/
@@ -276,3 +275,8 @@ tests/.ci_generated/
 
 # Local Lance precision-matching dump scripts (not part of the repo)
 lance_compare/
+
+# FP8 quantization test sample images: tests/diffusion/quantization/
+# test_quantization_fp8.py saves these into the repo root for manual
+# inspection; they are byproducts and must never be committed.
+test_*_fp8.png

README.md

@@ -16,7 +16,6 @@ Easy, fast, and cheap omni-modality model serving for everyone
 ---
 
 *Latest News* 🔥
-- [2026/07] We released [0.24.0](https://github.com/vllm-project/vllm-omni/releases/tag/v0.24.0) - aligned with the vLLM 0.24 release line, expanding production-ready coverage across TTS, speech, diffusion, image/video generation, and robot-policy serving, with major Omni stage runtime refactoring, diffusion request-level batching, async output materialization, quantization/cache/memory improvements, and broad CUDA/ROCm/XPU/NPU support.
 - [2026/06] We released [0.22.0](https://github.com/vllm-project/vllm-omni/releases/tag/v0.22.0) - an **omnimodal world-model** release aligned with vLLM 0.22, featuring [Nvidia Cosmos3](recipes/cosmos3/Cosmos3-Nano.md)/DreamZero world model support, expanded quantization coverage across Blackwell/NPU/XPU, TTS production improvements, new models including MiniCPM-o 4.5, MOSS-TTS, and Lance, plus RL integration with [VeRL-Omni](https://github.com/verl-project/verl-omni).
 - [2026/05] We released [0.20.0](https://github.com/vllm-project/vllm-omni/releases/tag/v0.20.0) - refreshes the serving/runtime stack for large-scale omni workloads, and improves diffusion model performance, quantization, and hardware readiness across CUDA, ROCm, MUSA, NPU, and XPU backends.
 - [2026/03] We released [0.18.0](https://github.com/vllm-project/vllm-omni/releases/tag/v0.18.0) - strengthens the core runtime through a large entrypoint refactor and scheduler/runtime cleanups, expands unified quantization and diffusion execution, broadens multimodal model coverage, and improves production readiness across audio, omni, image, video, RL, and multi-platform deployments.
@@ -32,9 +31,9 @@ Easy, fast, and cheap omni-modality model serving for everyone
 
 [vLLM](https://github.com/vllm-project/vllm) was originally designed to support large language models for text-based autoregressive generation tasks. vLLM-Omni is a framework that extends its support for omni-modality model inference and serving:
 
-- **Omni-modality**: Text, image, audio, video, and action data processing
+- **Omni-modality**: Text, image, video, and audio data processing
 - **Non-autoregressive Architectures**: extend the AR support of vLLM to Diffusion Transformers (DiT) and other parallel generation models
-- **Heterogeneous outputs**: from traditional text generation to multimodal and action outputs
+- **Heterogeneous outputs**: from traditional text generation to multimodal outputs
 
 <p align="center">
   <picture>
@@ -58,10 +57,9 @@ vLLM-Omni is flexible and easy to use with:
 
 vLLM-Omni seamlessly supports most popular open-source models on HuggingFace, including:
 
-- **Omni-modality models** (e.g. Qwen3-Omni, Cosmos3, HunyuanImage, BAGEL)
+- **Omni-modality models** (e.g. Qwen3-Omni, Cosmos, HunyuanImage, BAGEL)
 - **TTS models** (e.g. Qwen3-TTS, VoxCPM2, Ming-Omni-TTS, CosyVoice3)
 - **Diffusion models** — image, video, and audio generation (e.g. Qwen-Image, Wan2.2, FLUX)
-- **Robot-policy and action models** (e.g. GR00T-N1.7, DreamZero-DROID, InternVLA-A1, Cosmos3 action policy)
 
 ## Getting Started
 

benchmarks/diffusion/backends.py

@@ -28,7 +28,6 @@ class RequestFuncInput:
     slo_ms: float | None = None
     extra_body: dict[str, Any] = field(default_factory=dict)
     image_paths: list[str] | None = None
-    video_paths: list[str] | None = None
     request_id: str = field(default_factory=lambda: str(uuid.uuid4()))
     default_bot_task: str | None = DEFAULT_EDITS_BOT_TASK
 
@@ -339,11 +338,6 @@ async def async_request_v1_videos(
         form.add_field(k, str(v))
 
     image_file = None
-    if input.image_paths and input.video_paths:
-        output.error = "Only one of image_paths or video_paths can be provided"
-        output.success = False
-        return output
-
     if input.image_paths and len(input.image_paths) > 0:
         image_path = input.image_paths[0]
         image_file = open(image_path, "rb")
@@ -351,16 +345,7 @@ async def async_request_v1_videos(
             "input_reference",
             image_file,
             filename=os.path.basename(image_path),
-            content_type=_guess_mime_type(image_path),
-        )
-    elif input.video_paths and len(input.video_paths) > 0:
-        video_path = input.video_paths[0]
-        image_file = open(video_path, "rb")
-        form.add_field(
-            "input_reference",
-            image_file,
-            filename=os.path.basename(video_path),
-            content_type=_guess_mime_type(video_path),
+            content_type="application/octet-stream",
         )
 
     job_id = None

benchmarks/diffusion/diffusion_benchmark_serving.py

@@ -73,10 +73,6 @@ Usage:
             {"width":854,"height":480,"num_inference_steps":18,"num_frames":120,"fps":24,"weight":1}
         ]'
 
-    v2v:
-    python3 benchmarks/diffusion/diffusion_benchmark_serving.py \
-        --endpoint /v1/videos --dataset random --task v2v --num-prompts 1 \
-        --height 720 --width 1280 --fps 24 --num-frames 189 --num-inference-steps 35
 
 """
 
@@ -546,17 +542,11 @@ class TraceDataset(BaseDataset):
         if not image_paths:
             single = row.get("image_path")
             image_paths = [single] if single else None
-        video_paths = row.get("video_paths")
-        if not video_paths:
-            single_video = row.get("video_path")
-            video_paths = [single_video] if single_video else None
 
         if not image_paths and self.args.task in ["i2v", "i2i", "ti2v", "ti2i", "it2i"]:
             raise ValueError(
                 f"Task {self.args.task} requires image input, but no image_path or image_paths found in trace row."
             )
-        if not video_paths and self.args.task == "v2v":
-            raise ValueError("Task v2v requires video input, but no video_path or video_paths found in trace row.")
 
         override_w = self.args.width
         override_h = self.args.height
@@ -580,7 +570,6 @@ class TraceDataset(BaseDataset):
             timestamp=timestamp,
             slo_ms=slo_ms,
             image_paths=image_paths,
-            video_paths=video_paths,
             request_id=str(row.get("request_id")) if row.get("request_id") is not None else str(uuid.uuid4()),
         )
 
@@ -600,8 +589,6 @@ class CustomDataset(BaseDataset):
     - seed (optional): Random seed
     - image_paths (optional): List of input image paths for i2i/i2v/ti2i tasks
     - image_urls (optional): List of input image URLs (alternative to image_paths)
-    - video_paths (optional): List of input video paths for v2v tasks
-    - video_urls (optional): List of input video URLs (alternative to video_paths)
 
     Example JSONL for ti2i:
     {"prompt": "Add sunset lighting", "width": 1024, "height": 1024, "image_urls": ["https://example.com/image.jpg"]}
@@ -673,42 +660,6 @@ class CustomDataset(BaseDataset):
 
         return None
 
-    def _resolve_video_paths(self, item: dict) -> list[str] | None:
-        if "video_paths" in item and item["video_paths"]:
-            paths = item["video_paths"]
-            if isinstance(paths, str):
-                paths = [paths]
-
-            valid_paths = []
-            for path in paths:
-                if os.path.exists(path):
-                    valid_paths.append(path)
-                else:
-                    raise ValueError(f"Video file not found: {path}")
-
-            return valid_paths if valid_paths else None
-
-        if "video_urls" in item and item["video_urls"]:
-            urls = item["video_urls"]
-            if isinstance(urls, str):
-                urls = [urls]
-
-            downloaded_paths = []
-            for url in urls:
-                try:
-                    response = requests.get(url, timeout=30)
-                    response.raise_for_status()
-                    suffix = os.path.splitext(url)[1] or ".mp4"
-                    with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as tmp:
-                        tmp.write(response.content)
-                        downloaded_paths.append(tmp.name)
-                except Exception as e:
-                    raise ValueError(f"Failed to download video from {url}: {e}")
-
-            return downloaded_paths if downloaded_paths else None
-
-        return None
-
     def __len__(self) -> int:
         return len(self.data)
 
@@ -721,22 +672,11 @@ class CustomDataset(BaseDataset):
         height = item.get("height", self.args.height)
         num_inference_steps = item.get("num_inference_steps", self.args.num_inference_steps)
         seed = item.get("seed", self.args.seed)
-        reserved_keys = {
-            "prompt",
-            "width",
-            "height",
-            "num_inference_steps",
-            "seed",
-            "image_paths",
-            "image_urls",
-            "video_paths",
-            "video_urls",
-        }
+        reserved_keys = {"prompt", "width", "height", "num_inference_steps", "seed", "image_paths", "image_urls"}
         extra_body = {k: v for k, v in item.items() if k not in reserved_keys}
 
         # Handle image paths/URLs
         image_paths = self._resolve_image_paths(item)
-        video_paths = self._resolve_video_paths(item)
 
         return RequestFuncInput(
             prompt=prompt,
@@ -744,7 +684,6 @@ class CustomDataset(BaseDataset):
             model=self.model,
             seed=seed,
             image_paths=image_paths,
-            video_paths=video_paths,
             extra_body=extra_body,
             width=width,
             height=height,
@@ -798,10 +737,6 @@ class RandomDataset(BaseDataset):
             self._random_image_path = self._generate_random_image_paths()
         else:
             self._random_image_path = None
-        if self.args.task == "v2v":
-            self._random_video_path = self._generate_random_video_paths()
-        else:
-            self._random_video_path = None
 
     def __len__(self) -> int:
         return self.num_prompts
@@ -821,15 +756,13 @@ class RandomDataset(BaseDataset):
         if self._sampled_requests:
             profile = self._sampled_requests[idx]
             params.update(profile)
-        prompt = str(params.pop("prompt", f"Random prompt {idx} for benchmarking diffusion models"))
         return RequestFuncInput(
-            prompt=prompt,
+            prompt=f"Random prompt {idx} for benchmarking diffusion models",
             api_url=self.api_url,
             model=self.model,
             seed=self.args.seed,
             extra_body=extra_body,
             image_paths=self._random_image_path,
-            video_paths=self._random_video_path,
             **params,
         )
 
@@ -848,35 +781,6 @@ class RandomDataset(BaseDataset):
             image_paths.append(image_path)
         return image_paths
 
-    def _generate_random_video_paths(self) -> list[str]:
-        video_path = os.path.join(tempfile.gettempdir(), "diffusion_benchmark_random_video.mp4")
-        try:
-            import cv2
-
-            profile = self.random_request_config[0] if self.random_request_config else {}
-            width = int(self.args.width or profile.get("width") or 1280)
-            height = int(self.args.height or profile.get("height") or 720)
-            num_frames = int(self.args.num_frames or profile.get("num_frames") or 16)
-            fps = float(self.args.fps or profile.get("fps") or 8)
-            writer = cv2.VideoWriter(video_path, cv2.VideoWriter_fourcc(*"mp4v"), fps, (width, height))
-            if not writer.isOpened():
-                raise RuntimeError("cv2.VideoWriter failed to open")
-            for frame_idx in range(num_frames):
-                frame = np.zeros((height, width, 3), dtype=np.uint8)
-                x = int((frame_idx / max(num_frames - 1, 1)) * max(width - 96, 1))
-                y = height // 2
-                frame[:, :, 0] = 24
-                frame[:, :, 1] = 32
-                frame[:, :, 2] = 48
-                frame[max(y - 36, 0) : min(y + 36, height), x : min(x + 96, width), :] = (64, 128, 220)
-                writer.write(frame)
-            writer.release()
-            return [video_path]
-        except Exception as e:
-            raise RuntimeError(
-                "Failed to generate synthetic v2v input video. Install opencv-python or provide a custom dataset."
-            ) from e
-
 
 def _compute_expected_latency_ms_from_base(req: RequestFuncInput, args, base_time_ms: float | None) -> float | None:
     """Compute expected execution time (ms) based on a base per-step-per-frame unit time.
@@ -1208,7 +1112,7 @@ def wait_for_service(base_url: str, timeout: int = 120) -> None:
 
 
 def _default_endpoint_for_task(task: str) -> str:
-    if task in {"t2v", "i2v", "ti2v", "v2v"}:
+    if task in {"t2v", "i2v", "ti2v"}:
         return "/v1/videos"
     if task in {"i2i", "ti2i", "it2i"}:
         return "/v1/images/edits"
@@ -1222,7 +1126,7 @@ async def benchmark(args):
     if args.base_url is None:
         args.base_url = f"http://{args.host}:{args.port}"
 
-    VIDEO_TASKS = {"t2v", "i2v", "ti2v", "v2v"}
+    VIDEO_TASKS = {"t2v", "i2v", "ti2v"}
     IMAGE_TASKS = {"t2i", "i2i", "ti2i", "it2i"}
 
     if args.task in VIDEO_TASKS:
@@ -1273,10 +1177,6 @@ async def benchmark(args):
     if args.return_stage_metrics and args.endpoint in _STAGE_METRICS_ENDPOINTS:
         for req in requests_list:
             req.extra_body.setdefault(_RETURN_STAGE_METRICS_FIELD, True)
-    if args.extra_body:
-        extra_body = json.loads(args.extra_body)
-        for req in requests_list:
-            req.extra_body.update(extra_body)
 
     if args.endpoint == "/v1/images/edits":
         for req in requests_list:
@@ -1428,7 +1328,7 @@ if __name__ == "__main__":
         "--task",
         type=str,
         default="t2v",
-        choices=["t2v", "i2v", "ti2v", "v2v", "ti2i", "i2i", "it2i", "t2i"],
+        choices=["t2v", "i2v", "ti2v", "ti2i", "i2i", "it2i", "t2i"],
         help="Task type.",
     )
     parser.add_argument(
@@ -1543,12 +1443,6 @@ if __name__ == "__main__":
             '{"width":768,"height":768,"num_inference_steps":20,"weight":0.85}]'
         ),
     )
-    parser.add_argument(
-        "--extra-body",
-        type=str,
-        default=None,
-        help="JSON object merged into every generated request body/form.",
-    )
     parser.add_argument(
         "--num-input-images",
         type=int,

docker/Dockerfile.ci

@@ -2,7 +2,7 @@
 # The pinned VLLM_PRECOMPILED_WHEEL_COMMIT and VLLM_BASE_TAG below must
 # be updated on every rebase to match the new upstream vllm commit/tag.
 ARG VLLM_BASE_IMAGE=vllm/vllm-openai
-ARG VLLM_BASE_TAG=v0.25.0
+ARG VLLM_BASE_TAG=v0.24.0
 FROM ${VLLM_BASE_IMAGE}:${VLLM_BASE_TAG}
 ARG APP_DIR=/workspace/vllm-omni
 WORKDIR ${APP_DIR}
@@ -14,8 +14,71 @@ RUN apt-get update && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
+
+# Install vLLM (CUDA 13.0 build) from precompiled wheel at the selected commit.
+# Must use direct URL because the wheel has a PEP 440 local version identifier
+# (e.g. +gaeee7ef93.cu130) which pip/uv refuse to install from a PEP 503 package index.
+# The cu130 wheel index lives at /<commit>/cu130/vllm/ but the actual wheel files
+# are stored at the top level /<commit>/<wheel>.
+ENV VLLM_PRECOMPILED_WHEEL_COMMIT=dd10e03f95f94edbea1975c67ace3a35ec9a8a40
+# Keep the base image's CUDA-12 toolkit libs on the loader path. NOTE: this does
+# NOT resolve libnvJitLink.so.13 (that CUDA-13 lib is not in /usr/local/cuda here) —
+# the .so.13 JIT linker is handled separately via ldconfig after the cu130 wheels
+# are installed (see the nvidia-cuda-nvjitlink-cu13 step below).
+ENV LD_LIBRARY_PATH=/usr/local/cuda/targets/x86_64-linux/lib:${LD_LIBRARY_PATH}
+RUN VLLM_WHEEL_URL=$(python3 -c "import urllib.request,re; \
+    from urllib.parse import urljoin, quote; \
+    index_url='https://wheels.vllm.ai/${VLLM_PRECOMPILED_WHEEL_COMMIT}/cu130/vllm/'; \
+    html=urllib.request.urlopen(index_url).read().decode(); \
+    m=re.search(r'href=\"([^\"]*x86_64\.whl)\"',html); \
+    assert m, f'No x86_64 vLLM wheel found at {index_url}'; \
+    print(quote(urljoin(index_url,m.group(1)), safe=':/+%'))") && \
+    echo "Installing vLLM from: ${VLLM_WHEEL_URL}" && \
+    uv pip install --system --force-reinstall "${VLLM_WHEEL_URL}"
+
+
+
 RUN uv pip install --system ".[dev]"
 
+RUN uv pip install --system --upgrade \
+        "flashinfer-python==0.6.13" \
+        "flashinfer-cubin==0.6.13"
+
+RUN uv pip install --system --upgrade \
+    "flashinfer-jit-cache==0.6.13" \
+    --index-url https://flashinfer.ai/whl/cu130
+
+# Force-reinstall torch AFTER flashinfer, because flashinfer --upgrade pulls in
+# a newer torch (2.12.x) whose libtorch_cuda.so lacks
+# at::cuda::getCurrentCUDABlasHandle, causing "undefined symbol" at import time.
+RUN uv pip install --system --force-reinstall \
+    "torch==2.11.0" "torchaudio==2.11.0" "torchvision==0.26.0" \
+    --torch-backend cu130
+
+# CUDA 13 JIT linker: libnvJitLink.so.13 is provided by the nvidia-nvjitlink
+# package that the cu130 torch/vllm wheels above pull into site-packages (the
+# v0.24.0 base image's /usr/local/cuda toolkit is CUDA 12, so the LD_LIBRARY_PATH
+# above never resolves the .so.13). Its lib dir is not on the dynamic-loader path,
+# so bitsandbytes / torch.compile worker subprocesses fail to dlopen it. Locate the
+# already-installed lib and register its dir with ldconfig (system-wide, survives
+# subprocess env resets). Only fall back to the pip wheel (correct name:
+# nvidia-nvjitlink-cu13, NOT nvidia-cuda-nvjitlink-cu13, which does not exist) if
+# the torch deps did not bring it in.
+RUN NVJITLINK_LIB="$(find / -name 'libnvJitLink.so.13*' 2>/dev/null | head -1)" && \
+    if [ -z "$NVJITLINK_LIB" ]; then \
+        uv pip install --system "nvidia-nvjitlink-cu13" && \
+        NVJITLINK_LIB="$(find / -name 'libnvJitLink.so.13*' 2>/dev/null | head -1)"; \
+    fi && \
+    test -n "$NVJITLINK_LIB" && \
+    dirname "$NVJITLINK_LIB" > /etc/ld.so.conf.d/nvjitlink-cu13.conf && \
+    ldconfig && \
+    echo "Registered ${NVJITLINK_LIB} with ldconfig"
+
+# Re-pin numpy LAST. numba requires numpy <= 2.4, but the flashinfer and torch cu130
+# reinstalls above resolve numpy back up to 2.5, which breaks `import numba` at test
+# time ("Numba needs NumPy 2.4 or less"). This must be the final dependency op so
+# nothing can clobber it again.
+RUN uv pip install --system "numpy==2.2.6"
 
 RUN ln -sf /usr/bin/python3 /usr/bin/python
 

docker/Dockerfile.cuda

@@ -1,4 +1,4 @@
-ARG BASE_IMAGE=vllm/vllm-openai:v0.25.0
+ARG BASE_IMAGE=vllm/vllm-openai:v0.24.0
 FROM ${BASE_IMAGE}
 
 ARG COMMON_WORKDIR=/app

docker/Dockerfile.npu

@@ -1,5 +1,5 @@
 ARG VLLM_ASCEND_IMAGE=quay.io/atlas-ci/vllm-ascend
-ARG VLLM_ASCEND_TAG=v0.25.0
+ARG VLLM_ASCEND_TAG=v0.24.0
 FROM ${VLLM_ASCEND_IMAGE}:${VLLM_ASCEND_TAG}
 
 # WORKDIR /vllm-workspace/vllm

docker/Dockerfile.npu.a3

@@ -1,5 +1,5 @@
 ARG VLLM_ASCEND_IMAGE=quay.io/atlas-ci/vllm-ascend
-ARG VLLM_ASCEND_TAG=v0.25.0-a3
+ARG VLLM_ASCEND_TAG=v0.24.0-a3
 FROM ${VLLM_ASCEND_IMAGE}:${VLLM_ASCEND_TAG}
 
 # WORKDIR /vllm-workspace/vllm

docker/Dockerfile.npu.ci

@@ -1,5 +1,5 @@
 ARG VLLM_ASCEND_IMAGE=quay.io/atlas-ci/vllm-ascend
-ARG VLLM_ASCEND_TAG=v0.25.0
+ARG VLLM_ASCEND_TAG=v0.24.0
 FROM ${VLLM_ASCEND_IMAGE}:${VLLM_ASCEND_TAG}
 
 ARG APP_DIR=/vllm-workspace/vllm-omni

docker/Dockerfile.npu.ci.a3

@@ -1,5 +1,5 @@
 ARG VLLM_ASCEND_IMAGE=quay.io/atlas-ci/vllm-ascend
-ARG VLLM_ASCEND_TAG=v0.25.0-a3
+ARG VLLM_ASCEND_TAG=v0.24.0-a3
 FROM ${VLLM_ASCEND_IMAGE}:${VLLM_ASCEND_TAG}
 
 ARG APP_DIR=/vllm-workspace/vllm-omni

docker/Dockerfile.rocm

@@ -1,4 +1,4 @@
-ARG BASE_IMAGE=vllm/vllm-openai-rocm:v0.25.0
+ARG BASE_IMAGE=vllm/vllm-openai-rocm:v0.24.0
 FROM ${BASE_IMAGE} AS base
 
 # Declare a variable to know if we want to use the nightly build or the stable build.

docker/Dockerfile.xpu

@@ -88,7 +88,7 @@ ENV UV_EXTRA_INDEX_URL=${PIP_EXTRA_INDEX_URL}
 ENV UV_INDEX_STRATEGY="unsafe-best-match"
 ENV UV_LINK_MODE="copy"
 
-ARG VLLM_VERSION=v0.25.0
+ARG VLLM_VERSION=v0.24.0
 RUN git clone -b ${VLLM_VERSION} https://github.com/vllm-project/vllm
 WORKDIR /workspace/vllm
 

docs/.nav.yml

@@ -23,7 +23,7 @@ nav:
       - Image-To-Video: user_guide/examples/offline_inference/image_to_video.md
       - InternVLA-A1: user_guide/examples/offline_inference/internvla_a1.md
       - MammothModa2-Preview: user_guide/examples/offline_inference/mammothmodal2_preview.md
-      - MiMo-Audio: user_guide/examples/offline_inference/mimo_audio.md
+      - MiMo-Audio Offline Inference: user_guide/examples/offline_inference/mimo_audio.md
       - Qwen2.5-Omni: user_guide/examples/offline_inference/qwen2_5_omni.md
       - Qwen3-Omni: user_guide/examples/offline_inference/qwen3_omni.md
       - Text-To-Audio: user_guide/examples/offline_inference/text_to_audio.md
@@ -38,7 +38,7 @@ nav:
       - GLM-Image Online Serving: user_guide/examples/online_serving/glm_image.md
       - Image-To-Image: user_guide/examples/online_serving/image_to_image.md
       - Image-To-Video: user_guide/examples/online_serving/image_to_video.md
-      - MiMo-Audio: user_guide/examples/online_serving/mimo_audio.md
+      - Online serving Example of vLLM-Omni for MiMo-Audio: user_guide/examples/online_serving/mimo_audio.md
       - Qwen2.5-Omni: user_guide/examples/online_serving/qwen2_5_omni.md
       - Qwen3-Omni: user_guide/examples/online_serving/qwen3_omni.md
       - Text-To-Speech: user_guide/examples/online_serving/text_to_speech.md
@@ -80,7 +80,6 @@ nav:
       - Quantized KV Cache: user_guide/quantization/quantized_kvcache.md
       - FP8 W8A8: user_guide/quantization/fp8.md
       - Int8 W8A8: user_guide/quantization/int8.md
-      - BitsAndBytes W4: user_guide/quantization/bitsandbytes.md
       - MXFP8 W8A8: user_guide/quantization/mxfp8.md
       - MXFP4 W4A4: user_guide/quantization/mxfp4.md
       - ModelOpt: user_guide/quantization/modelopt.md

docs/README.md

@@ -28,9 +28,9 @@ Easy, fast, and cheap omni-modality model serving for everyone
 
 [vLLM](https://github.com/vllm-project/vllm) was originally designed to support large language models for text-based autoregressive generation tasks. vLLM-Omni is a framework that extends its support for omni-modality model inference and serving:
 
-- **Omni-modality**: Text, image, audio, video, and action data processing
+- **Omni-modality**: Text, image, video, and audio data processing
 - **Non-autoregressive Architectures**: extend the AR support of vLLM to Diffusion Transformers (DiT) and other parallel generation models
-- **Heterogeneous outputs**: from traditional text generation to multimodal and action outputs
+- **Heterogeneous outputs**: from traditional text generation to multimodal outputs
 
 <p align="center">
   <picture>
@@ -55,10 +55,8 @@ vLLM-Omni is flexible and easy to use with:
 
 vLLM-Omni seamlessly supports most popular open-source models on HuggingFace, including:
 
-- Omni-modality models (e.g. Qwen3-Omni, Cosmos3, HunyuanImage, BAGEL)
-- TTS models (e.g. Qwen3-TTS, VoxCPM2, Ming-Omni-TTS, CosyVoice3)
-- Diffusion models — image, video, and audio generation (e.g. Qwen-Image, Wan2.2, FLUX)
-- Robot-policy and action models (e.g. GR00T-N1.7, DreamZero-DROID, InternVLA-A1, Cosmos3 action policy)
+- Omni-modality models (e.g. Qwen2.5-Omni, Qwen3-Omni)
+- Multi-modality generation models (e.g. Qwen-Image)
 
 For more information, checkout the following:
 

docs/api/README.md

@@ -46,17 +46,17 @@ Engine classes for offline and online inference.
 - [vllm_omni.engine.PromptEmbedsPayload][]
 - [vllm_omni.engine.arg_utils.OmniEngineArgs][]
 - [vllm_omni.engine.async_omni_engine.AsyncOmniEngine][]
+- [vllm_omni.engine.mm_outputs.MultimodalCompletionOutput][]
+- [vllm_omni.engine.mm_outputs.MultimodalPayload][]
 - [vllm_omni.engine.orchestrator.Orchestrator][]
 - [vllm_omni.engine.orchestrator.OrchestratorRequestState][]
+- [vllm_omni.engine.output_modality.OutputModality][]
+- [vllm_omni.engine.output_modality.TensorAccumulationStrategy][]
+- [vllm_omni.engine.output_processor.MultimodalOutputProcessor][]
+- [vllm_omni.engine.output_processor.OmniRequestState][]
 - [vllm_omni.engine.stage_engine_core_client.StageEngineCoreClient][]
 - [vllm_omni.engine.stage_init_utils.StageMetadata][]
 - [vllm_omni.engine.stage_init_utils.StartedLlmStage][]
-- [vllm_omni.outputs.mm_outputs.MultimodalCompletionOutput][]
-- [vllm_omni.outputs.mm_outputs.MultimodalPayload][]
-- [vllm_omni.outputs.output_modality.OutputModality][]
-- [vllm_omni.outputs.output_modality.TensorAccumulationStrategy][]
-- [vllm_omni.outputs.output_processor.MultimodalOutputProcessor][]
-- [vllm_omni.outputs.output_processor.OmniRequestState][]
 
 ## Core
 

docs/cli/README.md

@@ -18,17 +18,11 @@ Specify the port:
 vllm serve Qwen/Qwen2.5-Omni-7B --omni --port 8091
 ```
 
-For a migrated model, load a custom deploy configuration with `--deploy-config`:
-
+If you have custom stage configs file, launch the server with command below
 ```bash
-vllm serve Qwen/Qwen2.5-Omni-7B --omni --deploy-config /path/to/deploy_config.yaml
+vllm serve Qwen/Qwen2.5-Omni-7B --omni --stage-configs-path /path/to/stage_configs_file
 ```
 
-The deprecated `--stage-configs-path` flag is retained for models that still use the legacy `stage_args` schema:
-
-```bash
-vllm serve ByteDance-Seed/BAGEL-7B-MoT --omni --stage-configs-path /path/to/legacy_stage_config.yaml
-```
 
 ## bench
 

docs/cli/bench/serve.md

@@ -2,7 +2,7 @@
 The vllm bench command launches the vLLM-Omni benchmark to evaluate the performance of multimodal models.
 
 ## Notes
-vLLM-Omni registers the `openai-chat-omni`, `openai-audio-speech`, `openai-image-edits-omni`, and `daily-omni` serving benchmark backends.
+We currently only support using the "openai-chat-omni" and "openai-image-edits-omni" backend.
 
 ## Basic Parameter Description
 You can use `vllm bench serve --omni --help=all` to get descriptions of all parameters. The commonly used parameters are described below:
@@ -10,7 +10,7 @@ You can use `vllm bench serve --omni --help=all` to get descriptions of all para
   Enable Omni (multimodal) mode, supporting multimodal inputs and outputs such as images, videos, and audio.
 
 - `--backend`
-  Specify the backend adapter. vLLM-Omni adds `openai-chat-omni`, `openai-audio-speech`, `openai-image-edits-omni`, and `daily-omni` to the upstream vLLM backend choices.
+  Specify the backend adapter as openai-chat-omni, using OpenAI Chat compatible API behavior as the protocol. Currently only openai-chat-omni is supported.
 
 - `--model`
   The model identifier to load, filled according to the models supported by vLLM-Omni.
@@ -55,7 +55,7 @@ You can use `vllm bench serve --omni --help=all` to get descriptions of all para
                     'Allowed metric names are "ttft", "tpot", "itl", "ttfc", "tpoc", "icl", '
                     '"tpop", "e2el", "audio_ttfp", "audio_rtf", "audio_duration". '
 
-- `--print-stage`
+- `-print-stage`
 Print per-stage benchmark metrics for --omni serving when stage metrics are returned by the server. Disabled by default.
 
 - `--save-result`
@@ -312,7 +312,7 @@ vllm bench serve \
   --ignore-eos \
   --percentile-metrics ttft,tpot,itl \
   --random-output-len 2 \
-  --extra-body '{"modalities": ["text"]}'
+  --extra_body '{"modalities": ["text"]}'
 ```
 
 If successful, you will see the following output:
@@ -390,7 +390,7 @@ vllm bench serve --omni \
   --ignore-eos \
   --percentile-metrics ttft,tpot,itl,e2el,audio_ttfp,audio_rtf,ttfc,tpoc,icl \
   --random-output-len 900 \
-  --extra-body '{"modalities": ["text","audio"]}' \
+  --extra_body '{"modalities": ["text","audio"]}' \
   --print-stage
 ```
 

docs/configuration/composable_parallel.md

@@ -92,12 +92,12 @@ Full pattern hierarchy: `TakeRank`, `Union`, `GatherDim`, `AllGather`, `StitchSp
 
 From highest to lowest, when the same axis is touched by multiple layers:
 
-1. **CLI overrides** (`--tensor-parallel-size`, `--stage-overrides`, `--omni-lb-policy`, etc.). A CLI value always wins; a warning is emitted whenever it overrides a strategy-declared axis (see `_reconcile_strategy_with_cli` in `vllm_omni/config/config_factory.py`).
+1. **CLI overrides** (`--stage_N_tensor_parallel_size`, `--stage-overrides`, `--omni-lb-policy`, etc.). A CLI value always wins; a warning is emitted whenever it overrides a strategy-declared axis (see `_reconcile_strategy_with_cli` in `vllm_omni/config/config_factory.py`).
 2. **Strategy YAML** (`--strategy-config`). Derived sizing is written onto the merged stages with *conflict-on-explicit* semantics — if the deploy YAML already set a value to something different, application raises `StrategyApplyError`.
 3. **Deploy YAML** (`--deploy-config` or the bundled `vllm_omni/deploy/<model_type>.yaml`).
 4. **Parser defaults.**
 
-After CLI overrides land, the device-layout guard re-runs (`check_device_layout`, called from `_reconcile_strategy_with_cli`) against the *effective* `tp * dp * pp * num_replicas` and the resolved `devices` string, so a `devices` value passed through `--stage-overrides` or a `--tensor-parallel-size` value that conflicts with the strategy's world size cannot slip past the pre-spawn check.
+After CLI overrides land, the device-layout guard re-runs (`check_device_layout`, called from `_reconcile_strategy_with_cli`) against the *effective* `tp * dp * pp * num_replicas` and the resolved `devices` string, so a `--devices` or `--tensor-parallel-size` that conflicts with the strategy's world size cannot slip past the pre-spawn check.
 
 ## Worked example
 
@@ -214,7 +214,7 @@ Strategy keys MUST be `model_stage` *names* (strings), the same labels stage con
     Use the `model_stage` name (e.g. `thinker`) instead.
 
 !!! warning "Mixing CLI overrides with a strategy on the same axis"
-    A strategy is meant to be the single writer for the axes it declares. If a CLI override (`--tensor-parallel-size 2`, `--stage-overrides '{"0": {"tensor_parallel_size": 2}}'`) sets the same axis, the **CLI value wins** and a warning is emitted from `_reconcile_strategy_with_cli`. The device-layout guard then re-runs against the effective layout, so a CLI override that breaks `tp * dp * pp * num_replicas` will still fail fast at config time.
+    A strategy is meant to be the single writer for the axes it declares. If a CLI override (`--stage_0_tensor_parallel_size`, `--stage-overrides '{"0": {...}}'`) sets the same axis, the **CLI value wins** and a warning is emitted from `_reconcile_strategy_with_cli`. The device-layout guard then re-runs against the effective layout, so a CLI override that breaks `tp * dp * pp * num_replicas` will still fail fast at config time.
 
 !!! warning "Passing `--strategy-config` against the legacy `--stage-configs-path`"
     Composable parallel only applies on the registry-based deploy path. If a model still resolves through the legacy `stage_args` YAML (loaded via `--stage-configs-path`), `--strategy-config` is silently inapplicable and emits:

docs/configuration/stage_configs.md

@@ -1,11 +1,11 @@
-# Deploy and legacy stage configurations
+# Stage configs for vLLM-Omni
 
-In vLLM-Omni, a model's `PipelineConfig` defines its fixed stage topology, while a deploy configuration controls how those stages run. Models that have not migrated to this split still use legacy stage configurations with a `stage_args` schema.
+In vLLM-Omni, the target model is separated into multiple stages, which are processed by different LLMEngines, DiffusionEngines or other types of engines. Depending on different types of stages, such as Autoregressive (AR) stage or Diffusion transformer (DiT) stage, each can choose corresponding schedulers, model workers to load with the Engines in a plug-in fashion.
 
 !!! note
     Default deploy config YAMLs (for example, `vllm_omni/deploy/qwen2_5_omni.yaml`, `vllm_omni/deploy/qwen3_omni_moe.yaml`, and `vllm_omni/deploy/qwen3_tts.yaml`) are bundled and loaded automatically when neither `--stage-configs-path` nor `--deploy-config` is provided — the model registry resolves the right pipeline + deploy YAML by `model_type`. The bundled defaults have been verified on 1xH100 for Qwen2.5-Omni and 2xH100 for Qwen3-Omni. Models that have not yet migrated to the new schema continue to use the legacy `vllm_omni/model_executor/stage_configs/<model>.yaml` files via `--stage-configs-path`.
 
-## Deploy configuration schema
+## New deploy schema reference
 
 The new deploy schema lives under `vllm_omni/deploy/` and is paired with a frozen `PipelineConfig` registered by the model's `pipeline.py`. Each deploy YAML has these top-level fields:
 
@@ -82,14 +82,14 @@ stages:
     input_connectors:  {from_stage_0: shm}
 ```
 
-### CLI flags
+### CLI flags introduced in this refactor
 
 | Flag | Description |
 |------|-------------|
-| `--deploy-config PATH` | Load a new-schema deploy YAML. It is mutually exclusive with `--stage-configs-path`. **Optional** — when omitted, the bundled `vllm_omni/deploy/<model_type>.yaml` is auto-loaded by the model registry. |
+| `--deploy-config PATH` | Load a new-schema deploy YAML. Takes precedence over `--stage-configs-path`. **Optional** — when omitted, the bundled `vllm_omni/deploy/<model_type>.yaml` is auto-loaded by the model registry. |
 | `--stage-overrides JSON` | Per-stage JSON overrides, e.g. `'{"0":{"gpu_memory_utilization":0.5}}'`. Per-stage values always win over global flags. |
 | `--async-chunk` / `--no-async-chunk` | Flip the deploy YAML's `async_chunk:` bool. Unset (default) leaves the YAML value in force. |
-| `--stage-configs-path` | **Deprecated.** Load a legacy `stage_args` YAML. New-schema YAMLs passed to this flag are auto-detected for compatibility, but should use `--deploy-config`. It is mutually exclusive with `--deploy-config` and will be removed in a future release. |
+| `--stage-configs-path` | **Deprecated.** Accepts legacy `stage_args` yamls and (auto-detected) new deploy yamls; emits a deprecation warning. Migrate to `--deploy-config`. To be removed in a follow-up PR. |
 
 ### Stage-Based CLI Paradigm
 
@@ -146,7 +146,7 @@ CUDA_VISIBLE_DEVICES=1 vllm serve Qwen/Qwen3-Omni-30B-A3B-Instruct --omni \
 
 From highest to lowest:
 
-1. Per-stage overrides (`--stage-overrides` JSON)
+1. Per-stage flags (`--stage-overrides` JSON, `--stage-<id>-<key>` if registered)
 2. Explicit global CLI flags (`--gpu-memory-utilization 0.85`, etc.)
 3. Platform section (`platforms.npu.stages`, etc.) on top of the base `stages:`
 4. Overlay YAML (via `base_config:`) on top of the base YAML
@@ -210,7 +210,7 @@ Effective config per stage after the merge:
 | 1 | `max_model_len` | `16384` | global CLI |
 | 2 | (all defaults) | — | base YAML (no overrides apply) |
 
-Therefore, as a core part of vLLM-Omni, a model's pipeline and deployment configurations have several main functions:
+Therefore, as a core part of vLLM-Omni, the stage configs for a model have several main functions:
 
 - Claim partition of stages and their corresponding class implementation in `model_executor/models`.
 - The disaggregated configuration for each stage and the communication topology among them.
@@ -219,16 +219,16 @@ Therefore, as a core part of vLLM-Omni, a model's pipeline and deployment config
 - Default input parameters.
 
 To override specific parameters, explicitly inject the customized configuration schema
-in both online and offline instantiation flows. Use the `--deploy-config` flag
-when loading a deploy configuration, reserving the `--stage-configs-path` parameter
+in both online and offline instantiation flows. Prioritize the `--deploy-config` flag
+when loading the new-schema deploy YAML schemas, reserving the `--stage-configs-path` parameter
 exclusively to maintain compatibility with legacy `stage_args` YAML constructs.
 
 Examples:
 
-For offline inference (assuming the necessary dependencies have been imported):
+For offline (Assume necessary dependencies have been imported):
 ```python
 model_name = "Qwen/Qwen2.5-Omni-7B"
-omni = Omni(model=model_name, deploy_config="/path/to/deploy_config.yaml")
+omni = Omni(model=model_name, stage_configs_path="/path/to/custom_stage_configs.yaml")
 ```
 
 For online serving:
@@ -242,14 +242,11 @@ Legacy online serving:
 vllm serve ByteDance-Seed/BAGEL-7B-MoT --omni --port 8091 --stage-configs-path /path/to/stage_configs_file
 ```
 !!! important
-    We are actively iterating on the definition of deployment configurations, and we welcome feedback from users and developers.
-
-## Legacy stage configuration reference
-
-The remainder of this page documents the deprecated `stage_args` schema for models that have not migrated to deploy configurations. The following historical Qwen2.5-Omni configuration illustrates that schema; current Qwen2.5-Omni deployments should use `vllm_omni/deploy/qwen2_5_omni.yaml` instead.
+    We are actively iterating on the definition of stage configs, and we welcome all feedbacks from both community users and developers to help us shape the development!
 
+Below is a specific example of stage_configs.yaml in Qwen2.5-omni.
 ```python
-# Historical stage configuration for Qwen2.5-Omni.
+# stage config for running qwen2.5-omni with AsyncOmniEngine + Orchestrator runtime.
 stage_args:
   - stage_id: 0 # mark the unique id for each stage
     runtime: # The disaggregated configuration
@@ -341,7 +338,7 @@ runtime:
 
 ```
 
-## Legacy stage configuration arguments
+## Stage Configuration Arguments
 
 Each stage in the `stage_args` list contains the following configuration options:
 

docs/contributing/DOCS_GUIDE.md

@@ -67,21 +67,6 @@ docs/
 └── stylesheets/         # Custom CSS
 ```
 
-## Naming Model Examples
-
-Offline inference and online serving examples for the same model use the same
-directory name and the shared display name in
-`examples/model_display_names.yml`. Use these title forms:
-
-- `# <Model>: Offline inference`
-- `# <Model>: Online serving`
-
-The documentation generator uses the full title for the page H1 and the shared
-display name alone for navigation, and fails the build if a mapped README uses
-a different H1. Keep checkpoint identifiers in commands and prose rather than
-in the display name, and do not rename an existing example directory solely to
-adjust its title because the directory defines its public documentation URL.
-
 ## Publishing Documentation
 
 ### GitHub Pages (Recommended)

docs/contributing/ci/test_guide.md

@@ -12,37 +12,12 @@ When running test cases, you may need to install the following dependencies:
 
 ```bash
 uv pip install ".[dev]"
-apt-get install -y espeak-ng jq
+apt-get install -y ffmpeg
 ```
 
 ## Running Tests
 Our test scripts use the pytest framework. First, please use `git clone https://github.com/vllm-project/vllm-omni.git` to download the vllm-omni source code. Then, in the root directory of vllm-omni, you can run the following commands in your local test environment to execute the corresponding test cases.
 
-### CI job runners (L2–L4): logs, timing, and timeouts
-
-[`tools/run_ready_jobs.sh`](https://github.com/vllm-project/vllm-omni/blob/main/tools/run_ready_jobs.sh), [`tools/run_merge_jobs.sh`](https://github.com/vllm-project/vllm-omni/blob/main/tools/run_merge_jobs.sh), and [`tools/nightly/run_nightly_jobs.sh`](https://github.com/vllm-project/vllm-omni/blob/main/tools/nightly/run_nightly_jobs.sh) share the same run-time behavior (via [`tools/run_jobs_common.sh`](https://github.com/vllm-project/vllm-omni/blob/main/tools/run_jobs_common.sh)):
-
-**Log layout** (default under `logs/ready_jobs`, `logs/merge_jobs`, or `logs/nightly_jobs`; override with `--log-dir`):
-
-| Path | Description |
-|------|-------------|
-| `jobs/*.sh` | Generated per-step bash wrappers |
-| `jobs/.job_timeouts` | Job key → `timeout_in_minutes` from Buildkite YAML (when set) |
-| `<job_key>.log` | Tee output for each job |
-| `timing_summary.log` | Per-job duration and total wall time after the run completes |
-
-**Timing:** while each job runs, stderr prints `finished in …` or `failed after …`. When all jobs finish, a summary is printed and written to `timing_summary.log`, for example:
-
-```
-=== Job timing summary ===
-  TTS_VoxCPM2_Test  25m 40s  OK
-  Omni_Qwen3-Omni_Test  8m 05s  TIMED OUT
-Total wall time: 33m 45s (2 jobs)
-Failed jobs: 1/2
-```
-
-**Timeouts:** if a Buildkite step defines `timeout_in_minutes`, the runner wraps that job with `timeout ${N}m` (aligned with CI). Steps without `timeout_in_minutes` rely on any inline `timeout …` already present in the extracted pytest command. On timeout the process exits with code `124` and the summary marks the job as `TIMED OUT`.
-
 === "L1 level"
 
     ```bash
@@ -53,109 +28,61 @@ Failed jobs: 1/2
 
 === "L2 level"
 
-    **Recommended:** run CI-aligned jobs from the repo root via [`tools/run_ready_jobs.sh`](https://github.com/vllm-project/vllm-omni/blob/main/tools/run_ready_jobs.sh). The script reads [`.buildkite/test-ready.yml`](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/test-ready.yml), generates per-step bash wrappers, runs pytest, and tees logs under `logs/ready_jobs/` (see [CI job runners](#ci-job-runners-l2l4-logs-timing-and-timeouts) for timing and timeout behavior). Requires `bash`, `python3`, and PyYAML (`pip install pyyaml`).
-
-    ```bash
-    # All L2 and L1 jobs (default: every pytest step in test-ready.yml)
-    bash tools/run_ready_jobs.sh
-
-    # Skip L1-style "Simple Test" steps in test-ready.yml (group or "Simple ·" labels)
-    bash tools/run_ready_jobs.sh --skip-simple
-    bash tools/run_ready_jobs.sh --skip-simple --model-type diffusion
-
-    # Preview extracted commands without running
-    bash tools/run_ready_jobs.sh --dry-run
-
-    # Filter by model area (OR semantics): omni | tts | diffusion | all
-    bash tools/run_ready_jobs.sh --model-type omni
-    bash tools/run_ready_jobs.sh --model-type tts,diffusion
-
-    # Match a Buildkite label substring (e.g. one model suite)
-    bash tools/run_ready_jobs.sh --model-type tts --label-substr "VoxCPM2"
-
-    ```
-
-    **Ad hoc:** run a single test file or marker expression manually:
-
     ```bash
     cd tests
-    pytest -s -v test_xxxx.py --run-level=core_model
-    pytest -s -v -m "core_model and distributed_cuda and L4" --run-level=core_model
+    pytest -s -v -m "core_model and not cpu" --run-level=core_model
     ```
+    If you only want to run a specific test case, you can use:
+    ```bash
+    pytest -s -v test_xxxx.py --run-level=core_model
+    ```
+    If you only want to run specific test cases on a particular platform, you can use:
+    ```bash
+    pytest -s -v -m "core_model and distributed_cuda and L4"  --run-level=core_model
+    ```
+    The latest test commands for various test suites can be found in the [pipeline](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/test-ready.yml).
 
 === "L3 level"
 
-    **Recommended:** run CI-aligned jobs from the repo root via [`tools/run_merge_jobs.sh`](https://github.com/vllm-project/vllm-omni/blob/main/tools/run_merge_jobs.sh). The script reads [`.buildkite/test-merge.yml`](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/test-merge.yml), generates per-step bash wrappers, runs pytest, and tees logs under `logs/merge_jobs/` (see [CI job runners](#ci-job-runners-l2l4-logs-timing-and-timeouts)). Requires `bash`, `python3`, and PyYAML (`pip install pyyaml`).
-
     ```bash
-    # All L3 and L1 jobs (default: every pytest step in test-merge.yml)
-    bash tools/run_merge_jobs.sh
-
-    # Skip L1-style "Simple Test" steps in test-merge.yml (group or "Simple ·" labels)
-    bash tools/run_merge_jobs.sh --skip-simple
-    bash tools/run_merge_jobs.sh --skip-simple --model-type omni
-
-    # Preview extracted commands without running
-    bash tools/run_merge_jobs.sh --dry-run
-
-    # Filter by model area (OR semantics): omni | tts | diffusion | all
-    bash tools/run_merge_jobs.sh --model-type diffusion
-    bash tools/run_merge_jobs.sh --model-type omni,tts
-
-    # Match a Buildkite label substring
-    bash tools/run_merge_jobs.sh --model-type diffusion --label-substr "Wan22"
-
+    pytest -s -v -m "advanced_model" --run-level=advanced_model
     ```
-
-    **Ad hoc:** run a single test file or marker expression manually:
-
+    If you only want to run a specific test case, you can use:
     ```bash
     pytest -s -v test_xxxx.py --run-level=advanced_model
-    pytest -s -v -m "advanced_model and distributed_cuda and L4" --run-level=advanced_model
     ```
+    If you only want to run specific test cases on a particular platform, you can use:
+    ```bash
+    pytest -s -v -m "advanced_model and distributed_cuda and L4"  --run-level=advanced_model
+    ```
+    The latest L3 test commands for various test suites can be found in the [pipeline](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/test-merge.yml).
 
 
 === "L4 level"
 
-    **Recommended:** run CI-aligned nightly jobs from the repo root via [`tools/nightly/run_nightly_jobs.sh`](https://github.com/vllm-project/vllm-omni/blob/main/tools/nightly/run_nightly_jobs.sh). The script reads [`.buildkite/test-nightly.yml`](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/test-nightly.yml), generates per-step bash wrappers, runs pytest (perf jobs first, then aggregates perf JSON into Excel), and tees logs under `logs/nightly_jobs/` (see [CI job runners](#ci-job-runners-l2l4-logs-timing-and-timeouts); `generate_nightly_perf_excel` is included in the timing summary). Requires `bash`, `python3`, and PyYAML (`pip install pyyaml`).
-
     ```bash
-    # All nightly jobs (default: test-type all, model-type all)
-    bash tools/nightly/run_nightly_jobs.sh
-
-    # Preview extracted commands without running
-    bash tools/nightly/run_nightly_jobs.sh --dry-run
-
-    # Test kind (--test-type, OR semantics): perf | acc | function | stability | local | all
-    bash tools/nightly/run_nightly_jobs.sh --test-type function
-    bash tools/nightly/run_nightly_jobs.sh --test-type perf,acc
-
-    # Model area (--model-type, OR semantics): omni | tts | diffusion | all
-    bash tools/nightly/run_nightly_jobs.sh --test-type function --model-type omni
-    bash tools/nightly/run_nightly_jobs.sh --test-type perf --model-type diffusion
-
-    # Stability scripts (tests/dfx/stability/scripts/) and local marker tests
-    bash tools/nightly/run_nightly_jobs.sh --test-type stability --model-type omni
-    bash tools/nightly/run_nightly_jobs.sh --test-type local --model-type tts
+    cd tests
+    pytest -s -v -m "full_model" --run-level=full_model
     ```
-
-    Perf steps (label contains ``Perf Test``) run before other jobs; Excel output is written under ``logs/``. Diffusion nightly steps may also appear in [test-nightly-diffusion.yml](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/test-nightly-diffusion.yml) on CI.
-
-    **Ad hoc:** run a single test file or marker expression manually:
+    If you only want to run a specific test case, you can use:
+    ```bash
+    pytest -s -v test_xxxx.py --run-level=full_model
+    ```
+    If you only want to run specific test cases on a particular platform, you can use:
+    ```bash
+    pytest -s -v -m "full_model and distributed_cuda and L4"  --run-level=full_model
+    ```
+    Note: To run performance tests (defaults to ``test_qwen_omni.json``; use ``--test-config-file tests/dfx/perf/tests/test_tts.json`` for TTS):
+    ```bash
+    pytest -s -v tests/dfx/perf/scripts/run_benchmark.py
+    ```
+    The latest L4 (nightly) test commands use the `full_model` marker and `--run-level full_model` (see [test-nightly.yml](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/test-nightly.yml) and [test-nightly-diffusion.yml](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/test-nightly-diffusion.yml)). Example:
 
     ```bash
     cd tests
-    pytest -s -v test_xxxx.py --run-level=full_model
-    pytest -s -v -m "full_model and distributed_cuda and L4" --run-level=full_model
     pytest -s -v -m "full_model and (omni or tts) and H100" --run-level=full_model
     ```
 
-    To run a standalone perf benchmark (defaults to ``test_qwen_omni.json``; use ``--test-config-file tests/dfx/perf/tests/test_tts.json`` for TTS):
-
-    ```bash
-    pytest -s -v tests/dfx/perf/scripts/run_benchmark.py
-    ```
-
 === "L5 level"
 
     L5 includes stability and reliability testing. Typical commands:

docs/contributing/model/adding_diffusion_model.md

@@ -638,7 +638,7 @@ mkdir -p examples/online_serving/your_model_name
 
 - Script: `examples/offline_inference/your_model_name/end2end.py`
   - Parse args like BAGEL (`--model`, `--modality`, optional `--image-path`, `--steps`, etc.)
-  - Use `from vllm_omni.entrypoints.omni import Omni` for both multi-stage and diffusion-only models
+  - Use `from vllm_omni.entrypoints.omni import Omni` (or `OmniDiffusion` if your model is diffusion-only)
   - Save outputs (images/audio/video/text) with deterministic filenames (e.g., `output_0_0.png`)
 - Doc: `examples/offline_inference/your_model_name/README.md`
   - Include at least one runnable command, e.g.:

docs/design/architecture_overview.md

@@ -13,7 +13,7 @@ This document outlines the architectural design for vLLM-Omni.
 
 The primary goal of the vLLM-Omni project is to build the fastest and easiest-to-use open-source Omni-Modality model inference & serving engine. vLLM-Omni extends the original vLLM, which was created to support large language models for text-based autoregressive (AR) generation tasks. vLLM-Omni is designed to support:
 
-* **Non-textual Output:** Enables the integration, efficient processing and output of various data types, including but not limited to, images, audio, video, and action trajectories, alongside text.
+* **Non-textual Output:** Enables the integration, efficient processing and output of various data types, including but not limited to, images, audio, and video, alongside text.
 * **Non-Autoregressive Structure:** Support model structure beyond autoregressive, especially Diffusion Transformer (DiT), which is widely used in visual and audio generation.
 * **Integration with vLLM Core:** Maintain compatibility and leverage existing vLLM key modules and optimizations where applicable.
 * **Extensibility:** Design a modular and flexible architecture that can easily accommodate new modalities, model architectures, and output formats.

docs/design/module/dit_module.md

@@ -470,7 +470,7 @@ class Attention(nn.Module):
                  role="self", role_category=None, ...):
         # Resolve backend for this role from the active OmniDiffusionConfig
         config = get_current_diffusion_config_or_none()
-        attention_config = config.diffusion_attention_config if config is not None else None
+        attention_config = config.attention_config if config is not None else None
         attn_backend_cls, spec = get_attn_backend_for_role(
             role=role,
             head_size=head_size,
@@ -493,7 +493,7 @@ These backends provide the **kernel implementations** for attention computation.
 
 #### Backend Selection Mechanism
 
-Selection is driven by `OmniDiffusionConfig.diffusion_attention_config`, which carries a global `default` plus a `per_role` map of `AttentionSpec` entries. A role string is resolved in this order:
+Selection is driven by `AttentionConfig` on `OmniDiffusionConfig`, which carries a global `default` plus a `per_role` map of `AttentionSpec` entries. A role string is resolved in this order:
 
 ```python
 def get_attn_backend_for_role(role, head_size, attention_config=None, role_category=None):
@@ -909,7 +909,7 @@ def initialize_model_parallel(
 
 ```
 1. User Request
-   └─> Omni.generate(prompt)
+   └─> OmniDiffusion.generate(prompt)
        └─> Prepare OmniDiffusionRequest
            └─> DiffusionEngine.step(requests)
 

docs/features/custom_pipeline.md

@@ -10,7 +10,7 @@ This guide demonstrates how to use the newly added features for extending vLLM-O
 Three main features enable custom pipeline extension:
 
 1. **`WorkerWrapperBase`**: A wrapper class that enables dynamic worker extension with custom functionality
-2. **`diffusion_load_format`**: A parameter that controls how diffusion models are loaded, including support for custom pipelines
+2. **`load_format`**: A parameter that controls how diffusion models are loaded, including support for custom pipelines
 3. **`CustomPipelineWorkerExtension`**: An extension class that enables pipeline re-initialization with custom implementations
 
 ## Features
@@ -27,15 +27,15 @@ Three main features enable custom pipeline extension:
 
 **Location:** `vllm_omni/diffusion/worker/diffusion_worker.py`
 
-### diffusion_load_format parameter
+### load_format Parameter
 
-The `diffusion_load_format` parameter controls the initial diffusion model load. The relevant values are:
+The `load_format` parameter controls how diffusion models are loaded. It supports the following values:
 
 - **`"default"`**: Standard model loading using the model registry (default behavior)
-- **`"dummy"`**: Skip the initial model load; use this with `custom_pipeline_args["pipeline_class"]` for the custom pipeline workflow below
-- **`"diffusers"`**: Load through the Hugging Face Diffusers adapter
+- **`"custom_pipeline"`**: Load a custom pipeline class specified by `custom_pipeline_name`
+- **`"dummy"`**: Skip model loading (useful for testing or when pipeline will be initialized separately)
 
-**Location:** `vllm_omni/diffusion/data.py` (`OmniDiffusionConfig`)
+**Location:** `vllm_omni/diffusion/model_loader/diffusers_loader.py`
 
 ### CustomPipelineWorkerExtension
 

docs/getting_started/installation/gpu.md

@@ -34,7 +34,7 @@ vLLM-Omni is a Python library that supports the following GPU variants. The libr
 
 ### Pre-built wheels
 
-Note: Pre-built wheels are currently available for vLLM-Omni 0.11.0rc1, 0.12.0rc1, 0.14.0rc1, 0.14.0, 0.16.0, 0.18.0, 0.20.0, 0.21.0, 0.22.0, 0.23.0, 0.24.0, and 0.25.0. If you need a newer unreleased revision, please [build from source](https://docs.vllm.ai/projects/vllm-omni/en/latest/getting_started/installation/gpu/#build-wheel-from-source).
+Note: Pre-built wheels are currently available for vLLM-Omni 0.11.0rc1, 0.12.0rc1, 0.14.0rc1, 0.14.0, 0.16.0, 0.18.0, 0.20.0, 0.21.0, 0.22.0, 0.23.0, and 0.24.0. If you need a newer unreleased revision, please [build from source](https://docs.vllm.ai/projects/vllm-omni/en/latest/getting_started/installation/gpu/#build-wheel-from-source).
 
 === "NVIDIA CUDA"
 

docs/getting_started/installation/gpu/cuda.inc.md

@@ -20,7 +20,7 @@ Therefore, it is recommended to install vLLM and vLLM-Omni with a **fresh new**
 
 vLLM-Omni is built based on vLLM. Please install it with command below.
 ```bash
-uv pip install vllm==0.25.0 --torch-backend=auto
+uv pip install vllm==0.24.0 --torch-backend=auto
 ```
 
 #### Installation of vLLM-Omni
@@ -39,13 +39,13 @@ uv pip install 'vllm-omni[demo]'
 # --8<-- [start:build-wheel-from-source]
 
 #### Installation of vLLM
-If you do not need to modify source code of vLLM, you can directly install the stable 0.25.0 release version of the library
+If you do not need to modify source code of vLLM, you can directly install the stable 0.24.0 release version of the library
 
 ```bash
-uv pip install vllm==0.25.0 --torch-backend=auto
+uv pip install vllm==0.24.0 --torch-backend=auto
 ```
 
-The 0.25.0 release of vLLM ships CUDA 13.0-compatible binaries by default. If you need a different CUDA variant or want to reuse an existing PyTorch installation, build vLLM from source instead.
+The 0.24.0 release of vLLM ships CUDA 13.0-compatible binaries by default. If you need a different CUDA variant or want to reuse an existing PyTorch installation, build vLLM from source instead.
 
 #### Installation of vLLM-Omni
 Since vllm-omni is rapidly evolving, it's recommended to install it from source
@@ -66,12 +66,12 @@ If you want to check, modify or debug with source code of vLLM, install the libr
 ```bash
 git clone https://github.com/vllm-project/vllm.git
 cd vllm
-git checkout v0.25.0
+git checkout v0.24.0
 ```
 Set up environment variables to get pre-built wheels. If there are internet problems, just download the whl file manually. And set `VLLM_PRECOMPILED_WHEEL_LOCATION` as your local absolute path of whl file.
 ```bash
-#For CUDA 13.0 (the default for v0.25.0; the wheel filename has no `+cu130` suffix)
-export VLLM_PRECOMPILED_WHEEL_LOCATION=https://github.com/vllm-project/vllm/releases/download/v0.25.0/vllm-0.25.0-cp38-abi3-manylinux_2_28_x86_64.whl
+#For CUDA 13.0 (the default for v0.24.0; the wheel filename has no `+cu130` suffix)
+export VLLM_PRECOMPILED_WHEEL_LOCATION=https://github.com/vllm-project/vllm/releases/download/v0.24.0/vllm-0.24.0-cp38-abi3-manylinux_2_28_x86_64.whl
 ```
 Install vllm with command below (If you have no existing PyTorch).
 ```bash
@@ -102,7 +102,7 @@ docker run --runtime nvidia --gpus 2 \
     --env "HF_TOKEN=$HF_TOKEN" \
     -p 8091:8091 \
     --ipc=host \
-    vllm/vllm-omni:v0.25.0 \
+    vllm/vllm-omni:v0.24.0 \
     vllm serve Qwen/Qwen3-Omni-30B-A3B-Instruct --omni --port 8091
 ```
 

docs/getting_started/installation/gpu/rocm.inc.md

@@ -13,7 +13,7 @@ vLLM-Omni current recommends the steps in under setup through Docker Images.
 
 vLLM-Omni is built based on vLLM. Please install it with command below.
 ```bash
-uv pip install vllm==0.25.0+rocm723 --extra-index-url https://wheels.vllm.ai/rocm/0.25.0/rocm723
+uv pip install vllm==0.24.0+rocm722 --extra-index-url https://wheels.vllm.ai/rocm/0.24.0/rocm722
 ```
 
 #### Installation of vLLM-Omni
@@ -34,13 +34,13 @@ uv pip install onnxruntime-rocm
 # --8<-- [start:build-wheel-from-source]
 
 #### Installation of vLLM
-If you do not need to modify source code of vLLM, you can directly install the stable 0.25.0 release version of the library
+If you do not need to modify source code of vLLM, you can directly install the stable 0.24.0 release version of the library
 
 ```bash
-uv pip install vllm==0.25.0+rocm723 --extra-index-url https://wheels.vllm.ai/rocm/0.25.0/rocm723
+uv pip install vllm==0.24.0+rocm722 --extra-index-url https://wheels.vllm.ai/rocm/0.24.0/rocm722
 ```
 
-The pre-built 0.25.0 vLLM wheel targets ROCm 7.2.3. If you need a different ROCm stack or want to reuse an existing PyTorch installation, build vLLM from source instead.
+The pre-built 0.24.0 vLLM wheel targets ROCm 7.2.2. If you need a different ROCm stack or want to reuse an existing PyTorch installation, build vLLM from source instead.
 
 #### Installation of vLLM-Omni
 Since vllm-omni is rapidly evolving, it's recommended to install it from source
@@ -58,7 +58,7 @@ If you want to check, modify or debug with source code of vLLM, install the libr
 ```bash
 git clone https://github.com/vllm-project/vllm.git
 cd vllm
-git checkout v0.25.0
+git checkout v0.24.0
 python3 -m pip install -r requirements/rocm.txt
 python3 setup.py develop
 ```
@@ -130,7 +130,7 @@ docker run --rm \
   -v ~/.cache/huggingface:/root/.cache/huggingface \
   --env "HF_TOKEN=$HF_TOKEN" \
   -p 8091:8091 \
-  vllm/vllm-omni-rocm:v0.25.0 \
+  vllm/vllm-omni-rocm:v0.24.0 \
   --model Qwen/Qwen3-Omni-30B-A3B-Instruct --omni --port 8091
 ```
 
@@ -149,7 +149,7 @@ docker run --rm -it \
   -v ~/.cache/huggingface:/root/.cache/huggingface \
   --env "HF_TOKEN=$HF_TOKEN" \
   --entrypoint bash \
-  vllm/vllm-omni-rocm:v0.25.0
+  vllm/vllm-omni-rocm:v0.24.0
 ```
 
 # --8<-- [end:pre-built-images]

docs/getting_started/installation/npu/npu.inc.md

@@ -9,8 +9,8 @@ The recommended way to use vLLM-Omni on NPU is through the vLLM-Ascend pre-built
 
 ```bash
 # vLLM-Ascend image
-# Atlas A2: quay.io/atlas-ci/vllm-ascend:v0.25.0
-# Atlas A3: quay.io/atlas-ci/vllm-ascend:v0.25.0-a3
+# Atlas A2: quay.io/atlas-ci/vllm-ascend:v0.24.0
+# Atlas A3: quay.io/atlas-ci/vllm-ascend:v0.24.0-a3
 docker run --rm \
     --name vllm-omni-npu \
     --shm-size=1g \
@@ -28,11 +28,11 @@ docker run --rm \
     -v /etc/ascend_install.info:/etc/ascend_install.info \
     -v /root/.cache:/root/.cache \
     -p 8000:8000 \
-    -it quay.io/atlas-ci/vllm-ascend:0.25.0-a3 bash
+    -it quay.io/atlas-ci/vllm-ascend:0.24.0-a3 bash
 
 # Inside the container, install vLLM-Omni from source
 cd /vllm-workspace
-git clone -b v0.25.0 https://github.com/vllm-project/vllm-omni.git
+git clone -b v0.24.0 https://github.com/vllm-project/vllm-omni.git
 cd vllm-omni
 pip install -v -e . --no-build-isolation
 # or VLLM_OMNI_TARGET_DEVICE=npu pip install -v -e .
@@ -84,8 +84,8 @@ Supported images as following.
 Here's an example deployment command that has been verified on 4 x NPUs:
 
 ```bash
-# Atlas A2: quay.io/ascend/vllm-omni:v0.25.0
-# Atlas A3: quay.io/ascend/vllm-omni:v0.25.0-a3
+# Atlas A2: quay.io/ascend/vllm-omni:v0.24.0
+# Atlas A3: quay.io/ascend/vllm-omni:v0.24.0-a3
 docker run --rm \
     --name vllm-omni-a3 \
     --shm-size=4g \
@@ -110,7 +110,7 @@ docker run --rm \
     -v /etc/ascend_install.info:/etc/ascend_install.info \
     -v ~/.cache:/root/.cache \
     -p 8091:8091 \
-    -it quay.io/ascend/vllm-omni:v0.25.0-a3 bash
+    -it quay.io/ascend/vllm-omni:v0.24.0-a3 bash
 ```
 
 !!! tip

docs/getting_started/quickstart.md

@@ -19,10 +19,10 @@ uv venv --python 3.12 --seed
 source .venv/bin/activate
 
 # On CUDA
-uv pip install vllm==0.25.0 --torch-backend=auto
+uv pip install vllm==0.24.0 --torch-backend=auto
 
 # On ROCm
-uv pip install vllm==0.25.0+rocm723 --extra-index-url https://wheels.vllm.ai/rocm/0.25.0/rocm723
+uv pip install vllm==0.24.0+rocm722 --extra-index-url https://wheels.vllm.ai/rocm/0.24.0/rocm722
 
 git clone https://github.com/vllm-project/vllm-omni.git
 cd vllm-omni
@@ -34,7 +34,7 @@ For additional installation methods — please see the [installation guide](inst
 !!! note
     It is important to install the same major & minor version of vLLM and vLLM Omni, otherwise things may not work as expected. If the versions are misaligned, you will see a warning when you import vLLM Omni.
 
-    If you are seeing strange behavior with the `vllm` command not handling the `--omni` flag correctly, you most likely have a version mismatch with vLLM < `0.25.0` and vLLM Omni `0.25.0`, as vLLM Omni no longer hijacks the vLLM entrypoint. Updating vLLM should resolve this issue.
+    If you are seeing strange behavior with the `vllm` command not handling the `--omni` flag correctly, you most likely have a version mismatch with vLLM < `0.24.0` and vLLM Omni `0.24.0`, as vLLM Omni no longer hijacks the vLLM entrypoint. Updating vLLM should resolve this issue.
 
 ## Offline Inference
 

docs/mkdocs/hooks/generate_examples.py

@@ -16,35 +16,8 @@ ROOT_DIR_RELATIVE = "../../../../.."
 EXAMPLE_DIR = ROOT_DIR / "examples"
 EXAMPLE_DOC_DIR = ROOT_DIR / "docs/user_guide/examples"
 NAV_FILE = ROOT_DIR / "docs/.nav.yml"
-MODEL_DISPLAY_NAMES_FILE = EXAMPLE_DIR / "model_display_names.yml"
 MAX_INLINE_MATERIAL_SIZE = 8 * 1024
 
-SERVING_MODE_TITLES = {
-    "offline_inference": "Offline inference",
-    "online_serving": "Online serving",
-}
-
-
-def load_model_display_names() -> dict[str, str]:
-    try:
-        with open(MODEL_DISPLAY_NAMES_FILE, encoding="utf-8") as f:
-            model_display_names = yaml.safe_load(f)
-    except (OSError, yaml.YAMLError) as exc:
-        raise ValueError(f"Failed to load MODEL_DISPLAY_NAMES_FILE at {MODEL_DISPLAY_NAMES_FILE}: {exc}") from exc
-
-    if not isinstance(model_display_names, dict) or not all(
-        isinstance(key, str) and isinstance(value, str) for key, value in model_display_names.items()
-    ):
-        raise ValueError(
-            "MODEL_DISPLAY_NAMES_FILE "
-            f"at {MODEL_DISPLAY_NAMES_FILE} must contain a YAML mapping "
-            "of string model IDs to string display names"
-        )
-    return model_display_names
-
-
-MODEL_DISPLAY_NAMES = load_model_display_names()
-
 
 def fix_case(text: str) -> str:
     subs = {
@@ -81,7 +54,6 @@ class Example:
         main_file (Path): The main file in the directory.
         other_files (list[Path]): list of other files in the directory.
         title (str): The title of the document.
-        nav_title (str): The concise title used in navigation.
 
     Methods:
         __post_init__(): Initializes the main_file, other_files, and title attributes.
@@ -106,16 +78,6 @@ class Example:
     def is_code(self) -> bool:
         return self.main_file.suffix != ".md"
 
-    @property
-    def model_display_name(self) -> str | None:
-        if self.category not in SERVING_MODE_TITLES:
-            return None
-        return MODEL_DISPLAY_NAMES.get(self.path.stem)
-
-    @property
-    def nav_title(self) -> str:
-        return self.model_display_name or self.title
-
     def determine_main_file(self) -> Path:
         """
         Determines the main file in the given path.
@@ -189,25 +151,13 @@ class Example:
         return [file for file in self.path.rglob("*") if is_other_file(file)]
 
     def determine_title(self) -> str:
-        model_display_name = self.model_display_name
         if not self.is_code:
             # Specify encoding for building on Windows
             with open(self.main_file, encoding="utf-8") as f:
                 first_line = f.readline().strip()
             match = re.match(r"^#\s+(?P<title>.+)$", first_line)
-            if model_display_name:
-                expected_title = f"{model_display_name}: {SERVING_MODE_TITLES[self.category]}"
-                actual_title = match.group("title") if match else first_line
-                if actual_title != expected_title:
-                    raise ValueError(
-                        f"Model example title mismatch in {self.main_file}: "
-                        f"expected '# {expected_title}', got {first_line!r}"
-                    )
-                return expected_title
             if match:
                 return match.group("title")
-        elif model_display_name:
-            raise ValueError(f"Mapped model example must use a Markdown README: {self.path}")
         return fix_case(self.path.stem.replace("_", " ").title())
 
     def fix_relative_links(self, content: str) -> str:
@@ -350,7 +300,7 @@ def update_nav_file(examples: list[Example]):
         for example in category_examples:
             doc_path = EXAMPLE_DOC_DIR / example.category / f"{example.path.stem}.md"
             rel_path = doc_path.relative_to(ROOT_DIR / "docs")
-            category_items.append({example.nav_title: rel_path.as_posix()})
+            category_items.append({example.title: rel_path.as_posix()})
 
         if category_items:
             # Format category name (e.g., "offline_inference" -> "Offline Inference")

docs/models/supported_models.md

@@ -34,7 +34,6 @@ th {
 | `QwenImageEditPlusPipeline` | Qwen-Image-Edit-2511 | `Qwen/Qwen-Image-Edit-2511` | ✅︎ | ✅︎ | ✅︎ | ✅︎ |
 | `GlmImagePipeline` | GLM-Image | `zai-org/GLM-Image` | ✅︎ | ✅︎ | | |
 | `ZImagePipeline` | Z-Image | `Tongyi-MAI/Z-Image-Turbo` | ✅︎ | ✅︎ | ✅︎ | ✅︎ |
-| `Krea2Pipeline` | Krea 2 (Raw + Turbo) | `krea/Krea-2-Raw`, `krea/Krea-2-Turbo` | ✅︎ | | | |
 | `WanPipeline` | Wan2.1-T2V, Wan2.2-T2V, Wan2.2-TI2V | `Wan-AI/Wan2.1-T2V-1.3B-Diffusers`, `Wan-AI/Wan2.1-T2V-14B-Diffusers`, `Wan-AI/Wan2.2-T2V-A14B-Diffusers`, `Wan-AI/Wan2.2-TI2V-5B-Diffusers` | ✅︎ | ✅︎ | ✅︎ | ✅︎ |
 | `WanImageToVideoPipeline` | Wan2.2-I2V | `Wan-AI/Wan2.2-I2V-A14B-Diffusers` | ✅︎ | ✅︎ | ✅︎ | ✅︎ |
 | `Cosmos3OmniDiffusersPipeline` | Cosmos3 T2I, T2V, I2V, V2V, T2V with sound, action policy | `nvidia/Cosmos3-Nano`, `nvidia/Cosmos3-Super` | ✅︎ | ✅︎ | ✅︎ | ✅︎ |

docs/serving/audio_generate_api.md

@@ -66,7 +66,7 @@ Content-Type: application/json
 |-----------|------|---------|-------------|
 | `input` | string | **required** | Text prompt describing the audio to generate |
 | `model` | string | server's model | Model to use (optional, should match server if specified) |
-| `response_format` | string | "wav" | Audio format: wav, mp3, flac, pcm, opus |
+| `response_format` | string | "wav" | Audio format: wav, mp3, flac, pcm, aac, opus |
 | `speed` | float | 1.0 | Playback speed (0.25 - 4.0) |
 
 #### Diffusion Parameters
@@ -90,6 +90,7 @@ Returns binary audio data with the appropriate `Content-Type` header:
 | `mp3` | `audio/mpeg` |
 | `flac` | `audio/flac` |
 | `pcm` | `audio/pcm` |
+| `aac` | `audio/aac` |
 | `opus` | `audio/opus` |
 
 ## Examples
@@ -284,7 +285,7 @@ Pydantic validation failure (e.g. invalid `response_format`, `speed` out of rang
     "detail": [
         {
             "type": "literal_error",
-            "msg": "Input should be 'wav', 'pcm', 'flac', 'mp3' or 'opus'",
+            "msg": "Input should be 'wav', 'pcm', 'flac', 'mp3', 'aac' or 'opus'",
             ...
         }
     ]

docs/serving/image_edit_api.md

@@ -252,7 +252,7 @@ Validation errors (missing required fields):
 
 ```bash
 # Check if server is responding
-curl http://localhost:8000/v1/images/edits \
+curl -X http://localhost:8000/v1/images/edit \
   -F "prompt='test'"
 ```
 

docs/serving/image_generation_api.md

@@ -138,7 +138,7 @@ Content-Type: application/json
 | `model` | string | server's model | Model to use (optional, should match server if specified) |
 | `n` | integer | 1 | Number of images to generate (1-10) |
 | `size` | string | model defaults | Image dimensions in WxH format (e.g., "1024x1024", "512x512") |
-| `response_format` | string | "b64_json" | Response format (`"b64_json"` or `"file"`) |
+| `response_format` | string | "b64_json" | Response format (only "b64_json" supported) |
 | `user` | string | null | User identifier for tracking |
 
 #### vllm-omni Extension Parameters

docs/serving/speech_api.md

@@ -109,7 +109,7 @@ Content-Type: application/json
 | `input` | string | **required** | The text to synthesize into speech |
 | `model` | string | server's model | Model to use (optional, should match server if specified) |
 | `voice` | string | "vivian" | Speaker name (e.g., vivian, ryan, aiden) |
-| `response_format` | string | "wav" | Audio format: wav, mp3, flac, pcm, opus |
+| `response_format` | string | "wav" | Audio format: wav, mp3, flac, pcm, aac, opus |
 | `speed` | float | 1.0 | Playback speed (0.25-4.0) |
 
 #### vLLM-Omni Extension Parameters

docs/user_guide/diffusion/attention_backends.md

@@ -83,7 +83,7 @@ When constructing `OmniDiffusionConfig` directly:
 from vllm_omni.diffusion.data import AttentionConfig, AttentionSpec, OmniDiffusionConfig
 
 config = OmniDiffusionConfig(
-    diffusion_attention_config=AttentionConfig(
+    attention_config=AttentionConfig(
         default=AttentionSpec(backend="FLASH_ATTN"),
         per_role={
             "cross": AttentionSpec(backend="TORCH_SDPA"),

docs/user_guide/diffusion/cpu_offload_diffusion.md

@@ -33,7 +33,7 @@ m = Omni(model="Wan-AI/Wan2.2-T2V-A14B-Diffusers", enable_cpu_offload=True)
 
 **CLI:**
 ```bash
-vllm serve Wan-AI/Wan2.2-T2V-A14B-Diffusers --omni --enable-cpu-offload
+vllm-omni serve diffusion Wan-AI/Wan2.2-T2V-A14B-Diffusers --enable-cpu-offload
 ```
 
 ### To Support a Model
@@ -82,33 +82,6 @@ mutual exclusion: when one group runs, the other moves to CPU.
 - Support single GPU only for now
 
 
-### Component offloading for split models (e.g. Cosmos3)
-
-Some models split their transformer into mutually-exclusive *components* that run
-in different phases of a single forward pass rather than as separate pipeline
-components -- e.g. Cosmos3's understanding (reasoner) component runs once per
-generation while the generation (generator) component runs every denoising step.
-Such models have no separate text encoder to swap against, so the transformer
-owns a small model-local offload path and wraps each phase with
-`with self._offload_context(name):`
-
-```python
-class Cosmos3VFMTransformer(nn.Module):
-    def forward(self, ...):
-        with self._offload_context("reasoner"):
-            ...  # understanding pass, runs once
-        with self._offload_context("generator"):
-            ...  # denoising pass, runs every step
-```
-
-Model-level offloading then keeps exactly one component GPU-resident at a time
-(the other on CPU), reusing the same `SequentialOffloadHook` `.to()` movers. The
-pipeline opts in by exposing `enable_omni_model_cpu_offload` (which drives the
-transformer's `enable_model_cpu_offload` and pins the VAE). Layerwise offloading
-works for these models too -- each component declares its own block container via
-`_layerwise_offload_blocks_attrs`.
-
-
 ## Layerwise (Blockwise) Offloading
 
 ### How It Works
@@ -146,10 +119,10 @@ m = Omni(model="Wan-AI/Wan2.2-I2V-A14B-Diffusers", enable_layerwise_offload=True
 **CLI:**
 ```bash
 # Text-to-video
-vllm serve Wan-AI/Wan2.2-T2V-A14B-Diffusers --omni --enable-layerwise-offload
+vllm-omni serve diffusion Wan-AI/Wan2.2-T2V-A14B-Diffusers --enable-layerwise-offload
 
 # Or image-to-video
-vllm serve Wan-AI/Wan2.2-I2V-A14B-Diffusers --omni --enable-layerwise-offload
+vllm-omni serve diffusion Wan-AI/Wan2.2-I2V-A14B-Diffusers --enable-layerwise-offload
 ```
 
 ### To Support a Model
@@ -172,9 +145,9 @@ class Flux2Transformer2DModel(nn.Module):
 ```
 
 ### Limitations
-- Cold start latency increases because offloaded components must be moved to CPU
-  during setup; layerwise offload may add extra weight consolidation and pinning
-  work.
+- Cold start latency increases because of
+    1) components are loaded to CPU first at the very first during initialization,
+    2) weight consolidation and pinning
 - Performance depends on compute cost and H2D bandwidth as well
 - Support single GPU only for now
 
@@ -205,18 +178,11 @@ Both strategies use vLLM-Omni's hook registry system (`HookRegistry` and `ModelH
 
 ```
 OffloadBackend (base class)
-├── ModelLevelOffloadBackend → uses SequentialOffloadHook (.to() swap)
-│                              (delegates to a pipeline's enable_omni_model_cpu_offload
-│                               for split models like Cosmos3)
+├── ModelLevelOffloadBackend → uses SequentialOffloadHook
 └── LayerWiseOffloadBackend → uses LayerwiseOffloadHook
 ```
 
-Factory function `get_offload_backend()` selects the appropriate backend based on
-configuration.
-
-For split models, `ModelLevelOffloadBackend.enable()` detects a pipeline's
-`enable_omni_model_cpu_offload` hook and delegates to it; Cosmos3 then swaps its
-reasoner/generator components inside the model forward pass.
+Factory function `get_offload_backend()` selects the appropriate backend based on configuration.
 
 
 ## Supported Models

docs/user_guide/diffusion/frame_interpolation.md

@@ -47,8 +47,7 @@ For supported Wan2.2 pipelines, the execution order is:
 1. Diffusion worker finishes denoising and decodes the raw video tensor.
 2. Worker-side model-specific post-processing runs.
 3. If frame interpolation is enabled, RIFE interpolates the decoded video
-   tensor on the worker side and records a FPS multiplier in
-   `metadata.video.video_fps_multiplier`.
+   tensor on the worker side and records a FPS multiplier in `custom_output`.
 4. The API server receives the already-interpolated video and only performs
    MP4 export.
 

docs/user_guide/diffusion_features.md

@@ -105,7 +105,7 @@ The following tables show which models support each feature:
 
 > Notes:
 
-> 1. CPU Offload has two methods: Module-wise (default for models with DiT + text encoder) and Layerwise. The tables below show **Layerwise support** only. Split models like Cosmos3 (no separate text encoder) swap their reasoner/generator components for module-wise offload; see the [CPU Offload Guide](diffusion/cpu_offload_diffusion.md).
+> 1. CPU Offload has two methods: Module-wise (default for models with DiT + text encoder) and Layerwise. The tables below show **Layerwise support** only.
 > 2. The **💾Quantization** column is collapsed for readability. See [Quantization Overview](quantization/overview.md) for per-method and per-model support details.
 
 ### ImageGen
@@ -121,7 +121,6 @@ The following tables show which models support each feature:
 | **GLM-Image**            |     ❌     |     ❌      |           ❌           |       ✅        |         ✅         |          ❌          |   ✅    |             ❌             |          ❌           |       ❌        |        ❌         |
 | **Hidream-I1-Full**        |     ❌     |     ❌      |           ❌           |       ❌        |         ✅         |          ❌          |   ❌    |             ❌             |          ❌           |       ❌        |        ❌         |
 | **HunyuanImage3**        |     ❌     |     ✅      |           ❌           |       ❌        |         ✅         |          ❌          |   ❌    |             ❌             |          ❌           |       ✅        |        ❌         |
-| **Krea 2**               |     ❌     |     ✅      |           ❌           |       ❌        |         ❌         |          ❌          |   ✅    |             ✅             |      ✅ (decode)      |       ❌        |        ❌         |
 | **LongCat-Image**        |     ✅     |     ✅      |           ✅           |       ✅        |         ✅         |          ❌          |   ❌    |             ✅             |          ❌           |       ❌        |        ❌         |
 | **LongCat-Image-Edit**   |     ✅     |     ✅      |           ✅           |       ✅        |         ✅         |          ❌          |   ❌    |             ✅             |          ❌           |       ❌        |        ❌         |
 | **MagiHuman**            |     ❌     |     ❌      |           ❌           |       ❓        |         ✅         |          ❌          |   ❌    |             ✅             |          ❌           |       ❌        |        ❌         |
@@ -144,8 +143,7 @@ The following tables show which models support each feature:
 > Notes:
 > 1. Nextstep_1(T2I) does not support cache acceleration methods such as TeaCache or Cache-DiT.
 > 2. `Tongyi-MAI/Z-Image-Turbo` and `SII-GAIR/daVinci-MagiHuman-Base-1080p` are distilled models with minimal NFEs; CFG-Parallel is not necessary.
-> 3. Cosmos3 T2I uses `Cosmos3OmniDiffusersPipeline` with `modalities=["image"]`. Model-level CPU offload swaps the nested UND reasoner and GEN generator pathways; layerwise offload remains available for blockwise GEN/UND offload.
-> 4. Krea 2 currently supports single-GPU inference plus LoRA, Cache-DiT, HSDP, CPU/layerwise offload, and VAE-patch-parallel (decode). TP/SP/CFG-Parallel are not yet wired. The few-step distilled (Turbo) checkpoint uses `is_distilled=true` (fixed timestep shift `mu=1.15`); generate at 2048x2048 by default with `num_inference_steps≈8` and `guidance_scale=0`. The Raw checkpoint uses 1024x1024, `num_inference_steps=28`, and `guidance_scale=4.5`.
+> 3. Cosmos3 T2I uses `Cosmos3OmniDiffusersPipeline` with `modalities=["image"]`. Model-level CPU offload is not supported; use layerwise offload.
 
 ### VideoGen
 

docs/user_guide/examples/offline_inference/bagel.md

@@ -257,7 +257,7 @@ python end2end.py --model ByteDance-Seed/BAGEL-7B-MoT \
 | :------- | :--- | :------ | :---------- |
 | `--deploy-config` | string | `None` | Path to deploy YAML (auto-detected if omitted) |
 | `--stage-configs-path` | string | `None` | [Deprecated] Legacy path to `stage_args` YAML; prefer `--deploy-config` |
-| `--worker-backend` | choice | `multi_process` | `multi_process` or `ray` |
+| `--worker-backend` | choice | `process` | `process` or `ray` |
 | `--ray-address` | string | `None` | Ray cluster address |
 | `--quantization` | string | `None` | Quantization method (e.g. `fp8`) |
 | `--log-stats` | flag | `False` | Enable statistics logging |

docs/user_guide/examples/offline_inference/glm_image.md

@@ -26,22 +26,18 @@ Stage 0 (AR Model)                Stage 1 (Diffusion)
 
 ```python
 from vllm_omni.entrypoints.omni import Omni
-from vllm_omni.inputs.data import OmniDiffusionSamplingParams
 
 if __name__ == "__main__":
     omni = Omni(model="zai-org/GLM-Image")
     outputs = omni.generate(
         "A photorealistic mountain landscape at sunset",
-        sampling_params_list=[
-            omni.default_sampling_params_list[0],
-            OmniDiffusionSamplingParams(
-                height=1024,
-                width=1024,
-                num_inference_steps=50,
-                guidance_scale=1.5,
-                seed=42,
-            ),
-        ],
+        sampling_params={
+            "height": 1024,
+            "width": 1024,
+            "num_inference_steps": 50,
+            "guidance_scale": 1.5,
+            "seed": 42,
+        },
     )
     outputs[0].request_output.images[0].save("output.png")
 ```
@@ -50,7 +46,6 @@ if __name__ == "__main__":
 
 ```python
 from vllm_omni.entrypoints.omni import Omni
-from vllm_omni.inputs.data import OmniDiffusionSamplingParams
 
 if __name__ == "__main__":
     omni = Omni(model="zai-org/GLM-Image")
@@ -61,16 +56,13 @@ if __name__ == "__main__":
                 "image": "input.png",
             },
         },
-        sampling_params_list=[
-            omni.default_sampling_params_list[0],
-            OmniDiffusionSamplingParams(
-                height=1024,
-                width=1024,
-                num_inference_steps=50,
-                guidance_scale=1.5,
-                seed=42,
-            ),
-        ],
+        sampling_params={
+            "height": 1024,
+            "width": 1024,
+            "num_inference_steps": 50,
+            "guidance_scale": 1.5,
+            "seed": 42,
+        },
     )
     outputs[0].request_output.images[0].save("output.png")
 ```

docs/user_guide/examples/offline_inference/mimo_audio.md

@@ -1,4 +1,4 @@
-# MiMo-Audio: Offline inference
+# MiMo-Audio Offline Inference
 
 Source <https://github.com/vllm-project/vllm-omni/tree/main/examples/offline_inference/mimo_audio>.
 

docs/user_guide/examples/offline_inference/qwen2_5_omni.md

@@ -1,4 +1,4 @@
-# Qwen2.5-Omni: Offline inference
+# Qwen2.5-Omni
 
 Source <https://github.com/vllm-project/vllm-omni/tree/main/examples/offline_inference/qwen2_5_omni>.
 

docs/user_guide/examples/offline_inference/qwen3_omni.md

@@ -1,4 +1,4 @@
-# Qwen3-Omni: Offline inference
+# Qwen3-Omni
 
 Source <https://github.com/vllm-project/vllm-omni/tree/main/examples/offline_inference/qwen3_omni>.
 

docs/user_guide/examples/online_serving/aura_omni.md

@@ -1,4 +1,4 @@
-# AURA Omni: Online serving
+# AURA Omni Pipeline
 
 `aura_omni` wires ASR, AURA, and Qwen3-TTS into one vLLM-Omni pipeline:
 

docs/user_guide/examples/online_serving/mimo_audio.md

@@ -1,4 +1,4 @@
-# MiMo-Audio: Online serving
+# Online serving Example of vLLM-Omni for MiMo-Audio
 
 Source <https://github.com/vllm-project/vllm-omni/tree/main/examples/online_serving/mimo_audio>.
 

docs/user_guide/examples/online_serving/qwen2_5_omni.md

@@ -1,4 +1,4 @@
-# Qwen2.5-Omni: Online serving
+# Qwen2.5-Omni
 
 Source <https://github.com/vllm-project/vllm-omni/tree/main/examples/online_serving/qwen2_5_omni>.
 

docs/user_guide/examples/online_serving/qwen3_omni.md

@@ -1,4 +1,4 @@
-# Qwen3-Omni: Online serving
+# Qwen3-Omni
 
 Source <https://github.com/vllm-project/vllm-omni/tree/main/examples/online_serving/qwen3_omni>.
 

docs/user_guide/examples/online_serving/text_to_audio.md

@@ -139,7 +139,7 @@ Content-Type: application/json
 |-----------|------|---------|-------------|
 | `input` | string | **required** | Text prompt describing the audio to generate |
 | `model` | string | server's model | Model to use (optional, should match server if specified) |
-| `response_format` | string | "wav" | Audio format: wav, mp3, flac, pcm, opus |
+| `response_format` | string | "wav" | Audio format: wav, mp3, flac, pcm, aac, opus |
 | `speed` | float | 1.0 | Playback speed (0.25 - 4.0) |
 | `audio_length` | float | null | Audio duration in seconds (max ~47s for `stable-audio-open-1.0`) |
 | `audio_start` | float | 0.0 | Audio start time in seconds |

docs/user_guide/quantization/autoround.md

@@ -3,7 +3,7 @@
 ## Overview
 
 [AutoRound](https://github.com/intel/auto-round) produces pre-quantized
-checkpoints for LLMs, VLMs, diffusion models, and world models. vLLM-Omni reads the
+checkpoints for LLMs, VLMs, and diffusion models. vLLM-Omni reads the
 checkpoint's `config.json` and auto-detects
 `quantization_config.quant_method = "auto-round"`.
 
@@ -36,13 +36,6 @@ guide. AutoRound is Intel-supported.
 | Wan2.2-T2V | `Intel/Wan2.2-T2V-A14B-Diffusers-int4-AutoRound` | Diffusion transformer | W4A16 | GPTQ-Marlin or Intel-supported AutoRound backend |
 | Wan2.2-TI2V | `Intel/Wan2.2-TI2V-5B-Diffusers-int4-AutoRound` | Diffusion transformer | W4A16 | GPTQ-Marlin or Intel-supported AutoRound backend |
 
-### World Model (Cosmos3)
-
-| Model | Checkpoint | Scope | Scheme | Backend |
-|-------|------------|-------|--------|---------|
-| Cosmos3-Nano | `Intel/Cosmos3-Nano-int4-AutoRound` | World-model transformer | W4A16 | GPTQ-Marlin or Intel-supported AutoRound backend |
-| Cosmos3-Super | `Intel/Cosmos3-Super-int4-AutoRound` | World-model transformer | W4A16 | GPTQ-Marlin or Intel-supported AutoRound backend |
-
 ### Multi-Stage Omni/TTS Model (Qwen3-Omni, Qwen3-TTS)
 
 | Model | Checkpoint | Scope | Scheme | Backend |

docs/user_guide/quantization/bitsandbytes.md

@@ -1,120 +0,0 @@
-# BitsAndBytes W4 Quantization
-
-## Overview
-
-BitsAndBytes 4-bit quantization supports weight-only NF4/FP4 diffusion transformer
-inference on CUDA GPUs. It quantizes BF16/FP16 weights at load time; activations
-stay in BF16/FP16. No pre-quantized checkpoint is required.
-
-This is an online (dynamic) path: load the normal HuggingFace checkpoint and
-quantize during model loading via the `bitsandbytes` CUDA kernels.
-
-## Hardware Support
-
-| Device | Support |
-|--------|---------|
-| NVIDIA CUDA GPU (SM 75+) | ✅ |
-| NVIDIA Blackwell GPU (SM 100+) | ✅ |
-| NVIDIA Ada/Hopper GPU (SM 89+) | ✅ |
-| NVIDIA Ampere GPU (SM 80+) | ✅ |
-| AMD ROCm | ❌ |
-| Intel XPU | ❌ |
-| Ascend NPU | ❌ |
-
-Legend: `✅` supported, `❌` unsupported. Non-CUDA platforms raise in
-`get_quant_method()`; CUDA requires compute capability 7.5+ (`SM 75+`).
-
-Requires the optional `bitsandbytes` package (`pip install bitsandbytes`).
-
-## Model Type Support
-
-### Diffusion Model (Qwen-Image, Wan2.2)
-
-| Model | HF models | CUDA | Mode | Recommendation |
-|-------|-----------|:----:|------|----------------|
-| Z-Image | `Tongyi-MAI/Z-Image-Turbo` | Yes | Online W4 weight-only | All heavy linear layers; sensitive embedders stay BF16 |
-| Qwen-Image | `Qwen/Qwen-Image`, `Qwen/Qwen-Image-2512` | Not validated | Online W4 weight-only | Compare vs BF16 before enabling |
-| Wan2.2 | Wan2.2 diffusion pipelines | Not validated | Online W4 weight-only | Validate before enabling in docs |
-
-Other diffusion models may work if their transformer uses supported linear
-layers, but they are not validated in this guide.
-
-### Multi-Stage Omni/TTS Model (Qwen3-Omni, Qwen3-TTS)
-
-| Model | Scope | Status | Notes |
-|-------|-------|--------|-------|
-| Qwen3-Omni | Thinker language-model stage | Not validated | Prefer checkpoint-supported ModelOpt FP8 or AutoRound paths |
-| Qwen3-TTS | TTS language-model stage | Not validated | No BitsAndBytes TTS stage support is documented |
-
-### Multi-Stage Diffusion Model (BAGEL, GLM-Image)
-
-| Model | Scope | Status | Notes |
-|-------|-------|--------|-------|
-| BAGEL | Stage-specific transformer or DiT module | Not validated | Requires explicit stage routing |
-| GLM-Image | Stage-specific transformer or DiT module | Not validated | Requires quality comparison with BF16 |
-
-## Configuration
-
-Python API:
-
-```python
-from vllm_omni import Omni
-from vllm_omni.inputs.data import OmniDiffusionSamplingParams
-
-omni = Omni(model="Tongyi-MAI/Z-Image-Turbo", quantization="bitsandbytes")
-
-omni_with_skips = Omni(
-    model="Tongyi-MAI/Z-Image-Turbo",
-    quantization_config={
-        "method": "bitsandbytes",
-        "ignored_layers": ["to_out"],
-    },
-)
-
-outputs = omni.generate(
-    "A cup of coffee on the table",
-    OmniDiffusionSamplingParams(num_inference_steps=50),
-)
-```
-
-CLI:
-
-```bash
-python text_to_image.py --model Tongyi-MAI/Z-Image-Turbo --quantization bitsandbytes
-python text_to_image.py --model Tongyi-MAI/Z-Image-Turbo --quantization bitsandbytes --ignored-layers "to_out"
-vllm serve Tongyi-MAI/Z-Image-Turbo --omni --quantization bitsandbytes
-```
-
-## Parameters
-
-| Parameter | Type | Default | Description |
-|-----------|------|---------|-------------|
-| `method` | str | - | Quantization method (`"bitsandbytes"`) |
-| `quant_type` | str | `"nf4"` | 4-bit data type: `"nf4"` (recommended) or `"fp4"` |
-| `compress_statistics` | bool | `True` | Double-quantize block scaling statistics for better accuracy |
-| `ignored_layers` | list[str] | `[]` | Layer name patterns to keep in BF16/FP16 |
-
-## Validation and Notes
-
-On Z-Image-Turbo (single GPU, 1024×1024, 50 steps), BitsAndBytes W4 typically
-reduces peak VRAM from roughly 24.5 GiB (BF16) to roughly 17 GiB. Compare output
-quality against a BF16 baseline before enabling on new models.
-
-Multi-GPU tensor parallelism (`tensor_parallel_size` > 1) is not validated for
-BitsAndBytes in diffusion models; each rank quantizes its own weight shard
-independently.
-
-If quality regresses, use `ignored_layers` to keep sensitive projections in BF16
-(for example `to_out` or `w2`).
-
-```python
-omni = Omni(
-    model="Tongyi-MAI/Z-Image-Turbo",
-    quantization="bitsandbytes",
-    cache_backend="tea_cache",
-    cache_config={"rel_l1_thresh": 0.2},
-)
-```
-
-Only add a new model to the supported table after comparing the BitsAndBytes
-output against a BF16 baseline and documenting any required `ignored_layers`.

docs/user_guide/quantization/gguf.md

@@ -9,15 +9,6 @@ for the transformer.
 
 GGUF is static quantization: the quantized weights are produced before serving.
 
-!!! note
-    GGUF support has migrated to OOT [vllm-gguf-plugin](https://github.com/vllm-project/vllm-gguf-plugin). Make sure you have GGUF plugin installed before serving a GGUF model.
-
-Before serving a GGUF model, make sure to install the [vllm-gguf-plugin](https://github.com/vllm-project/vllm-gguf-plugin):
-
-```bash
-uv pip install vllm-gguf-plugin>=0.0.3
-```
-
 ## Hardware Support
 
 | Device | Support |

examples/model_display_names.yml

@@ -1,12 +0,0 @@
-# Canonical display names for model examples that have both offline and online pages.
-# The docs generator combines these names with the serving mode for each page H1.
-aura_omni: AURA Omni
-covo_audio: Covo-Audio-Chat
-dynin_omni: Dynin-Omni
-lance: Lance
-mimo_audio: MiMo-Audio
-ming_flash_omni: Ming-flash-omni 2.0
-qwen2_5_omni: Qwen2.5-Omni
-qwen3_omni: Qwen3-Omni
-sensenova_u1: SenseNova-U1
-step_audio2: Step-Audio2

examples/offline_inference/aura_omni/README.md

@@ -1,4 +1,4 @@
-# AURA Omni: Offline inference
+# AURA Omni
 
 This example runs the native AURA Omni pipeline offline:
 

examples/offline_inference/covo_audio/README.md

@@ -1,4 +1,4 @@
-# Covo-Audio-Chat: Offline inference
+# Covo-Audio-Chat (Offline Inference)
 
 ## Setup
 

examples/offline_inference/dynin_omni/README.md

@@ -1,4 +1,4 @@
-# Dynin-Omni: Offline inference
+# Dynin-Omni Offline End2End Example
 
 This folder contains a unified offline inference entrypoint:
 

examples/offline_inference/image_to_video/image_to_video.py

@@ -220,7 +220,7 @@ def parse_args() -> argparse.Namespace:
         "--quantization",
         type=str,
         default=None,
-        choices=["fp8", "mxfp8", "mxfp4", "mxfp4_dualscale", "int8"],
+        choices=["fp8", "mxfp8", "mxfp4", "mxfp4_dualscale", "int8", "gguf"],
         help="Quantization method for the transformer. mxfp8: W8A8 MXFP8 (NPU). mxfp4: W4A4 MXFP4 (NPU). mxfp4_dualscale: W4A4 MXFP4 dual-scale + BF16 fallback mixed (NPU). fp8: online FP8 (GPU).",
     )
 

examples/offline_inference/lance/README.md

@@ -1,4 +1,4 @@
-# Lance: Offline inference
+# Lance (ByteDance) — offline inference
 
 [Lance](https://huggingface.co/bytedance-research/Lance) is a 3B unified
 autoregressive + diffusion multimodal model on a Qwen2.5-VL backbone. It is

examples/offline_inference/lance/end2end.py

@@ -293,19 +293,16 @@ def main():
 
     if args.modality in ("img2text", "video2text"):
         for i, req_output in enumerate(outputs):
-            multimodal_output = getattr(req_output, "multimodal_output", {}) or {}
-            metadata = multimodal_output.get("metadata", {}) if isinstance(multimodal_output, dict) else {}
-            text_metadata = metadata.get("text", {}) if isinstance(metadata, dict) else {}
-            text = (multimodal_output.get("text") if isinstance(multimodal_output, dict) else None) or (
-                text_metadata.get("text_output") if isinstance(text_metadata, dict) else None
-            )
-            text = text or getattr(req_output, "output", None) or getattr(req_output, "text", None)
+            text = getattr(req_output, "output", None) or getattr(req_output, "text", None)
+            if text is None and hasattr(req_output, "custom_output"):
+                text = (req_output.custom_output or {}).get("text_output")
             print(f"[Output {i}] {text}")
         return
 
     if args.modality in ("text2video", "video2video", "image2video"):
         for i, req_output in enumerate(outputs):
-            frames = getattr(req_output, "images", None)
+            custom = getattr(req_output, "custom_output", None) or {}
+            frames = custom.get("video_frames")
             if not frames:
                 print(f"[Output {i}] no video frames returned")
                 continue

examples/offline_inference/lance/gradio_demo.py

@@ -835,15 +835,12 @@ def run_task(
     print(f"[unified] task={task} {width}x{height} frames={num_frames} cfg={cfg} steps={steps} seed={seed}", flush=True)
 
     outputs = list(omni.generate(prompts=formatted, sampling_params_list=params))
-    multimodal_output = getattr(outputs[0], "multimodal_output", {}) or {}
-    metadata = multimodal_output.get("metadata", {}) if isinstance(multimodal_output, dict) else {}
-    text_metadata = metadata.get("text", {}) if isinstance(metadata, dict) else {}
+    custom = getattr(outputs[0], "custom_output", None) or {}
+    # Lance/Bagel return text at outputs[0].output (string) or
+    # custom_output["text_output"].
     text_out = None
     if task in {"x2t_image", "x2t_video"}:
-        text_out = (multimodal_output.get("text") if isinstance(multimodal_output, dict) else None) or (
-            text_metadata.get("text_output") if isinstance(text_metadata, dict) else None
-        )
-        text_out = text_out or getattr(outputs[0], "output", None)
+        text_out = custom.get("text_output") or getattr(outputs[0], "output", None)
         if not isinstance(text_out, str):
             text_out = None
 
@@ -859,11 +856,13 @@ def run_task(
             gr.update(value=str(text_out or "(empty)"), visible=True),
         )
     if task in {"t2v", "i2v", "video_edit"}:
-        frames = getattr(outputs[0], "images", None)
+        frames = custom.get("video_frames")
         if frames is None:
             raise gr.Error("No video frames in output.")
         return (gr.update(visible=False), gr.update(value=_save_video(frames), visible=True), gr.update(visible=False))
-    img_out = (getattr(outputs[0], "images", None) or [None])[0] or getattr(outputs[0], "output", None)
+    # image tasks — orchestrator strips ``outputs[0].output`` across the
+    # IPC boundary but preserves ``custom_output``, so prefer the latter.
+    img_out = custom.get("image") or getattr(outputs[0], "output", None)
     if img_out is None:
         raise gr.Error("No image in output.")
     return (gr.update(value=_save_image(img_out), visible=True), gr.update(visible=False), gr.update(visible=False))

examples/offline_inference/mimo_audio/README.md

@@ -1,4 +1,4 @@
-# MiMo-Audio: Offline inference
+# MiMo-Audio Offline Inference
 
 This directory contains an offline demo for running MiMo-Audio models with vLLM Omni. It builds task-specific inputs and generates WAV files or text outputs locally.
 

examples/offline_inference/ming_flash_omni/README.md

@@ -1,4 +1,4 @@
-# Ming-flash-omni 2.0: Offline inference
+# Ming-flash-omni 2.0
 
 [Ming-flash-omni-2.0](https://github.com/inclusionAI/Ming) is an omni-modal model supporting text, image, video, and audio understanding, with text and speech outputs.
 

examples/offline_inference/qwen2_5_omni/README.md

@@ -1,4 +1,4 @@
-# Qwen2.5-Omni: Offline inference
+# Qwen2.5-Omni
 
 ## Setup
 Please refer to the [stage configuration documentation](https://docs.vllm.ai/projects/vllm-omni/en/latest/configuration/stage_configs/) to configure memory allocation appropriately for your hardware setup.

examples/offline_inference/qwen3_omni/README.md

@@ -1,4 +1,4 @@
-# Qwen3-Omni: Offline inference
+# Qwen3-Omni
 
 ## Setup
 Please refer to the [stage configuration documentation](https://docs.vllm.ai/projects/vllm-omni/en/latest/configuration/stage_configs/) to configure memory allocation appropriately for your hardware setup.

examples/offline_inference/sensenova_u1/README.md

@@ -1,4 +1,4 @@
-# SenseNova-U1: Offline inference
+# SenseNova-U1-8B-MoT
 
 ## Architecture
 

examples/offline_inference/sensenova_u1/end2end.py

@@ -278,17 +278,12 @@ def main():
     )
 
     for req_output in outputs:
-        multimodal_output = getattr(req_output, "multimodal_output", {}) or {}
-        metadata = multimodal_output.get("metadata", {}) if isinstance(multimodal_output, dict) else {}
-        text_metadata = metadata.get("text", {}) if isinstance(metadata, dict) else {}
-        think_text = text_metadata.get("think_text") if isinstance(text_metadata, dict) else None
-        if args.print_think and think_text:
-            print(f"[Think]\n{think_text}\n")
+        custom = getattr(req_output, "_custom_output", {}) or {}
+        if args.print_think and custom.get("think_text"):
+            print(f"[Think]\n{custom['think_text']}\n")
 
         if is_text_output:
-            text = multimodal_output.get("text", "") if isinstance(multimodal_output, dict) else ""
-            if not text:
-                text = text_metadata.get("text_output", "") if isinstance(text_metadata, dict) else ""
+            text = custom.get("text_output", "")
             if not text:
                 text = getattr(req_output, "text", "") or ""
             print(f"[Response]\n{text}")

examples/offline_inference/step_audio2/README.md

@@ -1,4 +1,4 @@
-# Step-Audio2: Offline inference
+# Step-Audio2 Offline Inference Examples
 
 This directory contains examples for running offline inference with Step-Audio2 using vLLM-Omni.
 

examples/offline_inference/text_to_image/README.md

@@ -36,7 +36,6 @@ This folder provides several entrypoints for experimenting with text-to-image di
 | `black-forest-labs/FLUX.2-dev` | 1024 x 1024 | 65.7 | >80 (CPU offload required) |
 | `HunyuanImage-3.0` | 1024 x 1024 | 80.0 (TP≥3)  | 160 |
 | `HiDream-I1-Full` | 1024 x 1024 | 63.7  | 57.7 |
-| `krea/Krea-2-Raw`, `krea/Krea-2-Turbo` | 1024 (Raw) / 2048 (Turbo) | — | ~30 |
 
 !!! info
 *Peak VRAM:  based on basic single-card usage, batch size =1, without any acceleration/optimization features. FLUX.2-dev requires `--enable-cpu-offload` on a single 80 GiB GPU.
@@ -197,37 +196,6 @@ python examples/offline_inference/text_to_image/text_to_image.py \
   --output /output.png
 ```
 
-### Krea 2
-
-Krea 2 is published as diffusers-format checkpoints: `krea/Krea-2-Turbo` (few-step distilled) and `krea/Krea-2-Raw`.
-The pipeline class (`Krea2Pipeline`) is auto-detected from `model_index.json`. The distilled Turbo checkpoint sets
-`is_distilled=true` and is best run at 2048x2048 with few steps and guidance disabled:
-
-```bash
-# Few-step distilled (Turbo) checkpoint
-python examples/offline_inference/text_to_image/text_to_image.py \
-  --model krea/Krea-2-Turbo \
-  --prompt "a fox in the snow" \
-  --num-inference-steps 8 \
-  --guidance-scale 0.0 \
-  --height 2048 --width 2048 \
-  --output krea2.png
-```
-
-For the Raw checkpoint, use more steps with CFG enabled:
-
-```bash
-python examples/offline_inference/text_to_image/text_to_image.py \
-  --model krea/Krea-2-Raw \
-  --prompt "a fox in the snow" \
-  --num-inference-steps 28 \
-  --guidance-scale 4.5 \
-  --output krea2.png
-```
-
-`guidance-scale` follows the Krea 2 convention (`velocity = cond + guidance_scale * (cond - uncond)`); CFG is enabled
-whenever `guidance-scale > 0`.
-
 ### Batch Requests (Multiple Prompts)
 
 You can pass multiple prompts in a single `Omni.generate` call. `Omni`

examples/offline_inference/text_to_image/text_to_image.py

@@ -202,12 +202,17 @@ def parse_args() -> argparse.Namespace:
         "--quantization",
         type=str,
         default=None,
-        choices=["fp8", "int8", "bitsandbytes"],
+        choices=["fp8", "int8", "gguf"],
         help="Quantization method for the transformer. "
-        "Options: 'fp8' (FP8 W8A8 on Ada/Hopper, weight-only on older GPUs), "
-        "'int8' (Int8 W8A8), 'bitsandbytes' (NF4 4-bit weight-only, CUDA SM 75+). "
+        "Options: 'fp8' (FP8 W8A8 on Ada/Hopper, weight-only on older GPUs), 'int8' (Int8 W8A8), 'gguf' (GGUF quantized weights). "
         "Default: None (no quantization, uses BF16).",
     )
+    parser.add_argument(
+        "--gguf-model",
+        type=str,
+        default=None,
+        help=("GGUF file path or HF reference for transformer weights. Required when --quantization gguf is set."),
+    )
     parser.add_argument(
         "--ignored-layers",
         type=str,
@@ -397,7 +402,14 @@ def main():
     # ignored_layers is specified so the list flows through OmniDiffusionConfig
     quant_kwargs: dict[str, Any] = {}
     ignored_layers = [s.strip() for s in args.ignored_layers.split(",") if s.strip()] if args.ignored_layers else None
-    if args.quantization and ignored_layers:
+    if args.quantization == "gguf":
+        if not args.gguf_model:
+            raise ValueError("--gguf-model is required when --quantization gguf is set.")
+        quant_kwargs["quantization_config"] = {
+            "method": "gguf",
+            "gguf_model": args.gguf_model,
+        }
+    elif args.quantization and ignored_layers:
         quant_kwargs["quantization_config"] = {
             "method": args.quantization,
             "ignored_layers": ignored_layers,

examples/offline_inference/text_to_video/text_to_video.md

@@ -231,7 +231,7 @@ python text_to_video.py \
 - `--frame-rate`: generation FPS for pipelines that require it (e.g., LTX2).
 - `--audio-sample-rate`: audio sample rate for embedded audio (when the
   pipeline returns audio; LTX-2.3 outputs 48kHz audio).
-- `--quantization`: quantization method (such as `fp8` for FP8).
+- `--quantization`: quantization method (`fp8` for FP8, `gguf` for GGUF).
 - `--flow-shift`: scheduler flow_shift parameter.
 - `--extra-body`: JSON object of model-specific generation params, filtered against the model's declared `extra_body_params` (see [`vllm_omni/model_extras`](../../../vllm_omni/model_extras)). Used by Cosmos3 (see above).
 

examples/offline_inference/text_to_video/text_to_video.py

@@ -227,7 +227,7 @@ def parse_args() -> argparse.Namespace:
         "--quantization",
         type=str,
         default=None,
-        choices=["fp8", "mxfp8", "mxfp4", "mxfp4_dualscale", "int8"],
+        choices=["fp8", "mxfp8", "mxfp4", "mxfp4_dualscale", "int8", "gguf"],
         help="Quantization method for the transformer. mxfp8: W8A8 MXFP8 (NPU). mxfp4: W4A4 MXFP4 (NPU). mxfp4_dualscale: W4A4 MXFP4 dual-scale + BF16 fallback mixed (NPU). fp8: online FP8 (GPU).",
     )
 

examples/online_serving/aura_omni/README.md

@@ -1,4 +1,4 @@
-# AURA Omni: Online serving
+# AURA Omni Native Pipeline
 
 `aura_omni` serves AURA as a native multi-stage vLLM-Omni pipeline:
 

examples/online_serving/covo_audio/README.md

@@ -1,4 +1,4 @@
-# Covo-Audio-Chat: Online serving
+# Covo-Audio-Chat
 
 ## Installation
 

examples/online_serving/dynin_omni/README.md

@@ -1,4 +1,4 @@
-# Dynin-Omni: Online serving
+# Dynin-Omni Online Serving Example
 
 ## Installation
 

examples/online_serving/lance/README.md

@@ -1,4 +1,4 @@
-# Lance: Online serving
+# Lance — online serving
 
 OpenAI-compatible chat completions API for Lance, served by `vllm-omni`.
 

examples/online_serving/mimo_audio/README.md

@@ -1,4 +1,4 @@
-# MiMo-Audio: Online serving
+# Online serving Example of vLLM-Omni for MiMo-Audio
 
 ## 🛠️ Installation
 

examples/online_serving/ming_flash_omni/README.md

@@ -1,4 +1,4 @@
-# Ming-flash-omni 2.0: Online serving
+# Ming-flash-omni 2.0
 
 ## Installation
 

examples/online_serving/qwen2_5_omni/README.md

@@ -1,4 +1,4 @@
-# Qwen2.5-Omni: Online serving
+# Qwen2.5-Omni
 
 ## 🛠️ Installation
 

examples/online_serving/qwen3_omni/README.md

@@ -1,4 +1,4 @@
-# Qwen3-Omni: Online serving
+# Qwen3-Omni
 
 ## 🛠️ Installation
 

examples/online_serving/sensenova_u1/README.md

@@ -1,4 +1,4 @@
-# SenseNova-U1: Online serving
+# SenseNova-U1 Online Serving
 
 ## Launch the Server
 

examples/online_serving/step_audio2/README.md

@@ -1,4 +1,4 @@
-# Step-Audio2: Online serving
+# Step-Audio2 Online Serving
 
 This directory contains examples for running Step-Audio2 with vLLM-Omni's online serving API.
 

recipes/README.md

@@ -44,7 +44,6 @@ recipes/
 | [`inclusionAI/Ming-flash-omni-2.0.md`](./inclusionAI/Ming-flash-omni-2.0.md) | Online serving for multimodal chat + standalone TTS | 4x H100 / 1x H100 80GB |
 | [`inclusionAI/Ming-omni-tts.md`](./inclusionAI/Ming-omni-tts.md) | Offline + online dense Ming TTS/audio generation | 1x H100 80GB / 1x AMD MI300X (ROCm 7.2) |
 | [`IndexTeam/IndexTTS-2.md`](./IndexTeam/IndexTTS-2.md) | Online serving for voice-cloned TTS with optional emotion control | 1x L4 24GB or larger CUDA GPU |
-| [`krea/Krea-2.md`](./krea/Krea-2.md) | Text-to-image (Turbo + Raw), offline + online, with LoRA | 1x H100 80GB |
 | [`LTX/LTX-2.md`](./LTX/LTX-2.md) | Text-to-video and image-to-video serving | 1x H200 141GB |
 | [`LTX/LTX-2.3.md`](./LTX/LTX-2.3.md) | Text-to-video with audio generation (22B) | 1x GPU (96GB VRAM) |
 | [`MammothModa2/MammothModa2-Preview.md`](./MammothModa2/MammothModa2-Preview.md) | Text-to-image with the shared offline image example (AR → DiT) | 1x L40S 48GB / 1x ≥40GB GPU |

recipes/cosmos3/Cosmos3-Nano.md

@@ -422,185 +422,3 @@ ffprobe -v error -show_entries stream=codec_type,nb_frames,width,height cosmos3_
   `generate_sound`/`sound_duration`, `guardrails`, `action_*`, ...) are declared
   once in `vllm_omni/model_extras/cosmos3.py` and forwarded through `--extra-body`;
   unknown keys for the model are dropped.
-
-## NPU
-
-### 1x Ascend 910B / 910C (Atlas A2 / A3) — Online serving
-
-#### Environment
-
-- OS: Linux (aarch64)
-- Python: 3.12+
-- Driver / runtime: CANN 8.5.1 + NNAL + Ascend 910B / 910C
-- vLLM version: match the repository requirements from your current checkout
-- vLLM-Ascend version: match the repository requirements from your current checkout
-- vLLM-Omni version or commit: use the commit you are deploying from
-
-#### Command
-
-Requires the `vllm-omni` package (or the `quay.io/atlas-ci/vllm-ascend` A2 / A3 container),
-which provides the `vllm serve … --omni` entrypoint used below.
-
-Safety guardrails are **on by default** (NVIDIA Open Model License). They load
-the **gated** `nvidia/Cosmos-1.0-Guardrail` model, so to keep them on you must:
-
-1. `pip install cosmos-guardrail`
-2. Accept the license at <https://huggingface.co/nvidia/Cosmos-1.0-Guardrail>
-3. Export a token with access: `export HF_TOKEN=hf_...`
-
-Then launch the recommended server:
-
-```bash
-vllm serve nvidia/Cosmos3-Nano \
-  --omni \
-  --host 0.0.0.0 --port 8000 \
-  --init-timeout 1800
-```
-
-To run **without** guardrails (you are responsible for license compliance),
-add `--no-guardrails` (no token/`cosmos-guardrail` needed). For tensor parallel
-add `--tensor-parallel-size 8`. `--quantization fp8` and
-`--enable-layerwise-offload` are not supported on NPU.
-The pipeline auto-resolves from `model_index.json`; pass
-`--model-class-name Cosmos3OmniDiffusersPipeline` to force it explicitly.
-
-#### Verification
-
-Best quality uses the JSON-upsampled prompts from `assets/` (download with
-`hf download nvidia/Cosmos3-Nano assets/ --local-dir Cosmos3-Nano`). Minimal
-self-contained examples:
-
-```bash
-curl http://localhost:8000/v1/models
-
-# Text-to-image -> /v1/images/generations  (1024x1024, 10 steps; base64 PNG)
-curl -sS -X POST http://localhost:8000/v1/images/generations \
-  -H "Content-Type: application/json" \
-  -d '{
-    "model": "nvidia/Cosmos3-Nano",
-    "prompt": "A photorealistic red sports car on a city street at golden hour, cinematic lighting.",
-    "negative_prompt": "blurry, distorted, low quality",
-    "size": "1024x1024", "n": 1, "response_format": "b64_json",
-    "num_inference_steps": 10, "guidance_scale": 7.0, "seed": 42
-  }' | python -c "import sys,json,base64; open('cosmos3_t2i.png','wb').write(base64.b64decode(json.load(sys.stdin)['data'][0]['b64_json']))"
-
-# Text-to-video -> /v1/videos/sync  (720p, 49 frames @ 24fps)
-curl -sS -X POST http://localhost:8000/v1/videos/sync \
-  -H "Accept: video/mp4" \
-  -F "model=nvidia/Cosmos3-Nano" \
-  -F "prompt=A robot arm is cleaning a plate in the kitchen" \
-  -F "negative_prompt=blurry, distorted, low quality, jittery, deformed" \
-  -F "size=1280x720" -F "num_frames=49" -F "fps=24" \
-  -F "num_inference_steps=20" -F "guidance_scale=6.0" \
-  -F "max_sequence_length=4096" -F "flow_shift=10.0" \
-  -F "seed=123" \
-  -o cosmos3_t2v.mp4
-
-# Image-to-video -> /v1/videos/sync with an uploaded reference image
-curl -sS -X POST http://localhost:8000/v1/videos/sync \
-  -H "Accept: video/mp4" \
-  -F "model=nvidia/Cosmos3-Nano" \
-  -F "prompt=The scene comes to life with smooth, natural motion." \
-  -F "size=1280x720" -F "num_frames=25" -F "fps=8" \
-  -F "num_inference_steps=10" -F "guidance_scale=6.0" \
-  -F "seed=42" \
-  -F "input_reference=@reference.jpg;type=image/jpeg" \
-  -o cosmos3_i2v.mp4
-
-# Video-to-video -> /v1/videos/sync with an uploaded reference video
-curl -sS -X POST http://localhost:8000/v1/videos/sync \
-  -H "Accept: video/mp4" \
-  -F "model=nvidia/Cosmos3-Nano" \
-  -F "prompt=Continue the same scene with smooth natural motion and consistent subjects." \
-  -F "size=1280x720" -F "num_frames=17" -F "fps=5" \
-  -F "num_inference_steps=10" -F "guidance_scale=6.0" \
-  -F "seed=42" \
-  -F "input_reference=@reference.mp4;type=video/mp4" \
-  -o cosmos3_v2v.mp4
-```
-
-#### Notes
-
-- **Measured latency (1x Ascend 910B / 910C, bf16, guardrails off):**
-  - T2I 1024² — 10 steps → ~8 s
-  - T2V 1280×720 @ 20 steps — 49 frames → ~55 s
-  - I2V 1280×720 @ 10 steps — 25 frames → ~25 s
-  - V2V 480×320 @ 10 steps — 17 frames → ~12 s
-- **Memory:** transformer ~17 GiB (bf16); peak ~46 GiB for 720p video on 1 NPU;
-  full repo (transformer + Wan VAE + Qwen3-VL vision encoder + audio tokenizer)
-  ~33 GB on disk.
-- **Determinism:** identical seed reproduces identical output on the same
-  hardware; outputs are not bit-identical across different GPU/NPU types.
-- **Supported sizes (per model card):** 256p / 480p / 720p at 16:9, 4:3, 1:1,
-  3:4, 9:16. Defaults: T2I 1024², 50 steps, guidance 7.0; T2V/I2V/V2V
-  1280×720, 35 steps, guidance 6.0, `flow_shift=10.0`.
-- **Key flags / params:** `--no-guardrails` (optional, to disable guardrails), `--init-timeout 1800`
-  (for model loading), `--tensor-parallel-size 8` for multi-NPU, and
-  `--model-class-name Cosmos3OmniDiffusersPipeline` to force the pipeline class.
-- **Known limitations:**
-  - Transfer V2V with `extra_params` (`edge`/`blur`/`depth`/`seg`/`wsm`) hits a
-    resolution-parsing bug; basic V2V without transfer hints works.
-  - FP8 online quantization and layerwise offload are not supported on NPU.
-
-### 1x Ascend 910B / 910C (Atlas A2 / A3) — Offline generation
-
-#### Environment
-
-- OS: Linux (aarch64)
-- Python: 3.12+
-- Driver / runtime: CANN 8.5.1 + NNAL + Ascend 910B / 910C
-- vLLM-Omni version or commit: use the commit you are deploying from
-
-#### Command
-
-The same offline task examples run on NPU; pass model-specific knobs via
-`--extra-body`. Guardrails are on by default — pass `"guardrails": false` for a
-quick local run (install `cosmos-guardrail` + accept the gated repo to enable them).
-
-```bash
-# Text-to-image -> examples/offline_inference/text_to_image
-python examples/offline_inference/text_to_image/text_to_image.py \
-  --model nvidia/Cosmos3-Nano \
-  --prompt "A photorealistic red sports car at golden hour, cinematic lighting." \
-  --negative-prompt "blurry, distorted, low quality" \
-  --height 1024 --width 1024 --num-inference-steps 50 --guidance-scale 7.0 \
-  --extra-body '{"flow_shift": 3.0, "guardrails": false}' \
-  --output cosmos3_t2i.png
-
-# Text-to-video -> examples/offline_inference/text_to_video
-python examples/offline_inference/text_to_video/text_to_video.py \
-  --model nvidia/Cosmos3-Nano \
-  --prompt "A robot arm is cleaning a plate in the kitchen." \
-  --negative-prompt "blurry, distorted, low quality, jittery, deformed" \
-  --height 720 --width 1280 --num-frames 189 --fps 24 \
-  --num-inference-steps 35 --guidance-scale 6.0 \
-  --extra-body '{"flow_shift": 10.0, "max_sequence_length": 4096, "guardrails": false,
-                 "use_resolution_template": false, "use_duration_template": false}' \
-  --output cosmos3_t2v.mp4
-
-# Image-to-video -> examples/offline_inference/image_to_video
-# (Cosmos3 bundles example frames under assets/; any RGB image works too.)
-python examples/offline_inference/image_to_video/image_to_video.py \
-  --model nvidia/Cosmos3-Nano \
-  --image /path/to/Cosmos3-Nano/assets/example_i2v_input.jpg \
-  --prompt "The scene comes to life with smooth, natural motion." \
-  --height 720 --width 1280 --num-frames 189 --fps 24 \
-  --num-inference-steps 35 --guidance-scale 6.0 \
-  --extra-body '{"flow_shift": 10.0, "max_sequence_length": 4096, "guardrails": false}' \
-  --output cosmos3_i2v.mp4
-```
-
-#### Verification
-
-```bash
-python -c "from PIL import Image; im=Image.open('cosmos3_t2i.png'); print('image', im.size, im.mode)"
-ffprobe -v error -show_entries stream=codec_type,nb_frames,width,height cosmos3_t2v.mp4
-```
-
-#### Notes
-
-- Guardrails are on by default on NPU with `cosmos-guardrail` installed. Pass
-  `"guardrails": false` in `--extra-body` to disable them (there is no
-  `--no-guardrails` flag for the offline scripts).
-- Video at the 189-frame default takes ~15 min/clip on 1 NPU; reduce
-  `--num-frames` for faster iteration.

recipes/krea/Krea-2.md

@@ -1,143 +0,0 @@
-# Krea 2 for text-to-image (base + LoRA)
-
-## Summary
-
-- Vendor: Krea
-- Model: `krea/Krea-2-Turbo` (few-step distilled) and `krea/Krea-2-Raw`
-- Task: Text-to-image (T2I)
-- Mode: Offline engine and online OpenAI-compatible serving
-- Maintainer: Community
-
-## When to use this recipe
-
-Use this recipe for a known-good starting point to run the Krea 2 single-stream
-MMDiT with vLLM-Omni on a **single 80 GB NVIDIA H100** — both the offline
-`text_to_image.py` entrypoint and the online `/v1/images/generations` route,
-including PEFT LoRA adapters. `Krea2Pipeline` is auto-detected from
-`model_index.json`; guidance is checkpoint-aware (distilled Turbo runs no-CFG,
-Raw runs CFG).
-
-## References
-
-- Related example under `examples/`:
-  [`examples/offline_inference/text_to_image/README.md`](../../examples/offline_inference/text_to_image/README.md)
-  (see the "Krea 2" section)
-- Feature support:
-  [`docs/user_guide/diffusion_features.md`](../../docs/user_guide/diffusion_features.md)
-- Related PR: [#4730](https://github.com/vllm-project/vllm-omni/pull/4730)
-
-## Hardware Support
-
-This recipe documents a **single-GPU** CUDA layout on H100 80 GB. Add more
-platforms (ROCm / NPU) or multi-GPU layouts as community validation lands.
-
-## GPU
-
-### 1× H100 80GB
-
-#### Environment
-
-Versions from a working editable install (activate `vllm-omni/.venv` or your equivalent):
-
-- OS: Linux
-- Python: 3.12
-- Driver / runtime: NVIDIA driver **595.71.05** (CUDA 13.x)
-- torch: **2.11.0+cu130**
-- vLLM: **0.24.0**
-- vLLM-Omni: editable install from this repo (Git **`960ce8f6`**)
-- Transformers: **5.12.1**
-
-Krea 2 weights are ~30 GB and fit comfortably on a single 80 GB H100.
-
-#### Command
-
-Offline — few-step distilled (Turbo) checkpoint (2048×2048, no CFG):
-
-```bash
-python examples/offline_inference/text_to_image/text_to_image.py \
-  --model krea/Krea-2-Turbo \
-  --prompt "a fox in the snow" \
-  --num-inference-steps 8 \
-  --guidance-scale 0.0 \
-  --height 2048 --width 2048 \
-  --output krea2_turbo.png
-```
-
-Offline — Raw checkpoint (1024×1024, CFG enabled):
-
-```bash
-python examples/offline_inference/text_to_image/text_to_image.py \
-  --model krea/Krea-2-Raw \
-  --prompt "a fox in the snow" \
-  --num-inference-steps 28 \
-  --guidance-scale 4.5 \
-  --height 1024 --width 1024 \
-  --output krea2_raw.png
-```
-
-LoRA (offline) — pass a PEFT adapter with `--lora-path` / `--lora-scale`.
-`NagaSaiAbhinay/Krea-2-vllm-darkbrush-LoRA` is a vLLM-Omni-compatible repackaging
-of `krea/Krea-2-LoRA-darkbrush`; its trigger word is `monochrome ink wash style`:
-
-```bash
-python examples/offline_inference/text_to_image/text_to_image.py \
-  --model krea/Krea-2-Turbo \
-  --prompt "a fox in the snow, monochrome ink wash style" \
-  --num-inference-steps 8 --guidance-scale 0.0 \
-  --lora-path NagaSaiAbhinay/Krea-2-vllm-darkbrush-LoRA --lora-scale 1.0 \
-  --output krea2_darkbrush.png
-```
-
-Online — serve and query the OpenAI-compatible image route:
-
-```bash
-vllm serve krea/Krea-2-Turbo --omni --enforce-eager --port 8091
-```
-
-```bash
-curl -s http://localhost:8091/v1/images/generations \
-  -H "Content-Type: application/json" \
-  -d '{
-    "prompt": "a fox in the snow, monochrome ink wash style",
-    "size": "1024x1024",
-    "num_inference_steps": 8,
-    "guidance_scale": 0.0,
-    "seed": 42,
-    "lora": {"name": "darkbrush", "path": "NagaSaiAbhinay/Krea-2-vllm-darkbrush-LoRA", "scale": 1.0}
-  }' | jq -r '.data[0].b64_json' | base64 -d > krea2_online.png
-```
-
-Cache-DiT acceleration (offline) — add `--cache-backend cache_dit`:
-
-```bash
-python examples/offline_inference/text_to_image/text_to_image.py \
-  --model krea/Krea-2-Turbo --prompt "a fox in the snow" \
-  --num-inference-steps 8 --guidance-scale 0.0 \
-  --cache-backend cache_dit --output krea2_cachedit.png
-```
-
-#### Verification
-
-```bash
-# Offline: the script prints "Saved generated image to <path>"; confirm the file exists.
-ls -lh krea2_turbo.png
-```
-
-For the online route, decode the returned `b64_json` (see the `curl` above) and
-confirm a valid 1024×1024 PNG is written.
-
-#### Notes
-
-- Key flags: distilled Turbo → few steps (~8) with `guidance_scale=0.0`; Raw →
-  more steps (~28) with `guidance_scale>0` (CFG). `guidance-scale` follows the
-  Krea 2 convention `velocity = cond + guidance_scale * (cond - uncond)`.
-- LoRA: adapters must be PEFT format (`adapter_config.json` +
-  `adapter_model.safetensors`); the projections are `ReplicatedLinear`
-  (264 modules). Offline uses `--lora-path` (which sets
-  `OmniDiffusionSamplingParams.lora_request` / `lora_scale`); online uses the
-  top-level `lora` object. Style LoRAs like darkbrush need their trigger word in
-  the prompt.
-- Cache-DiT is supported; `has_separate_cfg` is checkpoint-aware (False for Turbo
-  no-CFG, True for Raw CFG).
-- Known limitations: TP / SP / CFG-Parallel are not yet wired for Krea 2. HSDP,
-  layerwise CPU offload, and VAE-patch-parallel (decode) are supported.

tests/ar_diffusion/test_engine_routing.py

@@ -134,17 +134,3 @@ def test_runner_rejects_multi_sequence_config():
     fake = SimpleNamespace(od_config=SimpleNamespace(max_num_seqs=4))
     with pytest.raises(ValueError, match="max_num_seqs=1"):
         ARDiffusionModelRunner._preallocate_kv_cache(fake)
-
-
-def test_execute_model_accepts_kv_prefetch_jobs_kwarg():
-    """Regression: the worker passes ``kv_prefetch_jobs`` to every runner
-    (added with the KV prefetch path, #4448); the AR-Diffusion override must
-    accept and forward it or DreamZero crashes on the warm-up dummy run."""
-    import inspect
-
-    from vllm_omni.diffusion.worker.diffusion_model_runner import DiffusionModelRunner
-    from vllm_omni.experimental.ar_diffusion.runner import ARDiffusionModelRunner
-
-    for cls in (DiffusionModelRunner, ARDiffusionModelRunner):
-        params = inspect.signature(cls.execute_model).parameters
-        assert "kv_prefetch_jobs" in params, cls

tests/ar_diffusion/test_pipeline_bridge.py

@@ -203,7 +203,7 @@ def test_failed_forward_tears_down_session(monkeypatch):
     runner.device = None
     runner._perf_e2e_times = []
 
-    def boom(self, req, kv_prefetch_jobs=None):
+    def boom(self, req):
         raise RuntimeError("layer 17 exploded")
 
     monkeypatch.setattr(DiffusionModelRunner, "execute_model", boom)

tests/config/test_omni_config.py

@@ -751,14 +751,3 @@ def test_diffusion_config_field_classification_covers_current_fields():
         "distributed_executor_backend",
     } <= omni_config_module._DIFFUSION_SHARED_CONFIG_FIELDS
     assert "prompt_file_path" in omni_config_module._DIFFUSION_RUNTIME_CONFIG_FIELDS
-
-
-def test_diffusion_config_projection_keeps_mapping_quantization_config_serializable():
-    quantization_config = {
-        "method": "example_quant",
-        "weights": "weights.bin",
-    }
-
-    cfg = omni_config_module._DiffusionConfigProjection.from_kwargs(quantization_config=quantization_config)
-
-    assert cfg.quantization_config == quantization_config

tests/dfx/perf/scripts/run_diffusion_benchmark.py

@@ -36,7 +36,7 @@ import pytest
 
 from benchmarks.diffusion.backends import endpoint_filename_token, normalize_endpoint
 
-pytestmark = [pytest.mark.diffusion, pytest.mark.full_model, pytest.mark.local_model]
+pytestmark = [pytest.mark.diffusion, pytest.mark.full_model]
 
 os.environ["VLLM_WORKER_MULTIPROC_METHOD"] = "spawn"
 os.environ.setdefault("DIFFUSION_ATTENTION_BACKEND", "FLASH_ATTN")
@@ -807,7 +807,7 @@ def assert_result(
 
 def _default_benchmark_endpoint_for_task(task: str) -> str:
     """Return the default client-side benchmark endpoint for a diffusion task."""
-    if task in {"t2v", "i2v", "ti2v", "v2v"}:
+    if task in {"t2v", "i2v", "ti2v"}:
         return "/v1/videos"
     if task in {"t2i", "i2i", "ti2i"}:
         return "/v1/chat/completions"

tests/dfx/perf/tests/test_cosmos3_vllm_omni.json

@@ -1,213 +0,0 @@
-[
-    {
-        "test_name": "test_cosmos3_t2i_official_demo_2gpu",
-        "description": "Cosmos3-Nano text-to-image official demo workload on 2 GPUs",
-        "server_type": "vllm-omni",
-        "benchmark_endpoint": "/v1/images/generations",
-        "server_params": {
-            "model": "nvidia/Cosmos3-Nano",
-            "serve_args": {
-                "cfg-parallel-size": 2,
-                "use-hsdp": true,
-                "hsdp-shard-size": 2,
-                "vae-patch-parallel-size": 2,
-                "vae-use-tiling": true,
-                "model-class-name": "Cosmos3OmniDiffusersPipeline",
-                "no-guardrails": true,
-                "stage-init-timeout": 900,
-                "init-timeout": 1200,
-                "enable-diffusion-pipeline-profiler": true
-            }
-        },
-        "benchmark_params": [
-            {
-                "name": "1024x1024_steps4",
-                "dataset": "random",
-                "task": "t2i",
-                "width": 1024,
-                "height": 1024,
-                "num-inference-steps": 4,
-                "num-prompts": 3,
-                "max-concurrency": 1,
-                "seed": 42,
-                "extra-body": {
-                    "guidance_scale": 7.0,
-                    "flow_shift": 3.0,
-                    "guardrails": false
-                },
-                "baseline": {
-                    "throughput_qps": 1.107,
-                    "latency_mean": 0.8929,
-                    "peak_memory_mb_mean": 80000.0
-                }
-            }
-        ]
-    },
-    {
-        "test_name": "test_cosmos3_t2v_official_demo_2gpu",
-        "description": "Cosmos3-Nano text-to-video official demo workload on 2 GPUs",
-        "server_type": "vllm-omni",
-        "benchmark_endpoint": "/v1/videos",
-        "server_params": {
-            "model": "nvidia/Cosmos3-Nano",
-            "serve_args": {
-                "cfg-parallel-size": 2,
-                "use-hsdp": true,
-                "hsdp-shard-size": 2,
-                "vae-patch-parallel-size": 2,
-                "vae-use-tiling": true,
-                "model-class-name": "Cosmos3OmniDiffusersPipeline",
-                "no-guardrails": true,
-                "stage-init-timeout": 900,
-                "init-timeout": 1200,
-                "enable-diffusion-pipeline-profiler": true
-            }
-        },
-        "benchmark_params": [
-            {
-                "name": "1280x720_frames189_steps4",
-                "dataset": "random",
-                "task": "t2v",
-                "num-prompts": 3,
-                "max-concurrency": 1,
-                "seed": 42,
-                "random-request-config": [
-                    {
-                        "width": 1280,
-                        "height": 720,
-                        "prompt": "A robot arm is cleaning a plate in the kitchen.",
-                        "num_inference_steps": 4,
-                        "num_frames": 189,
-                        "fps": 24,
-                        "weight": 1
-                    }
-                ],
-                "extra-body": {
-                    "guidance_scale": 6.0,
-                    "flow_shift": 10.0,
-                    "max_sequence_length": 4096,
-                    "guardrails": false,
-                    "use_resolution_template": false,
-                    "use_duration_template": false
-                },
-                "baseline": {
-                    "throughput_qps": 0.027,
-                    "latency_mean": 36.6824,
-                    "peak_memory_mb_mean": 30155.4
-                }
-            }
-        ]
-    },
-    {
-        "test_name": "test_cosmos3_i2v_official_demo_2gpu",
-        "description": "Cosmos3-Nano image-to-video official demo workload on 2 GPUs",
-        "server_type": "vllm-omni",
-        "benchmark_endpoint": "/v1/videos",
-        "server_params": {
-            "model": "nvidia/Cosmos3-Nano",
-            "serve_args": {
-                "cfg-parallel-size": 2,
-                "use-hsdp": true,
-                "hsdp-shard-size": 2,
-                "vae-patch-parallel-size": 2,
-                "vae-use-tiling": true,
-                "model-class-name": "Cosmos3OmniDiffusersPipeline",
-                "no-guardrails": true,
-                "stage-init-timeout": 900,
-                "init-timeout": 1200,
-                "enable-diffusion-pipeline-profiler": true
-            }
-        },
-        "benchmark_params": [
-            {
-                "name": "1280x720_frames189_steps4",
-                "dataset": "random",
-                "task": "i2v",
-                "num-prompts": 3,
-                "max-concurrency": 1,
-                "num-input-images": 1,
-                "seed": 42,
-                "random-request-config": [
-                    {
-                        "width": 1280,
-                        "height": 720,
-                        "prompt": "The scene comes to life with smooth, natural motion.",
-                        "num_inference_steps": 4,
-                        "num_frames": 189,
-                        "fps": 24,
-                        "weight": 1
-                    }
-                ],
-                "extra-body": {
-                    "guidance_scale": 6.0,
-                    "flow_shift": 10.0,
-                    "max_sequence_length": 4096,
-                    "guardrails": false
-                },
-                "baseline": {
-                    "throughput_qps": 0.027,
-                    "latency_mean": 36.968,
-                    "peak_memory_mb_mean": 30443.6
-                }
-            }
-        ]
-    },
-    {
-        "test_name": "test_cosmos3_v2v_official_demo_2gpu",
-        "description": "Cosmos3-Nano video-to-video official demo workload on 2 GPUs",
-        "server_type": "vllm-omni",
-        "benchmark_endpoint": "/v1/videos",
-        "server_params": {
-            "model": "nvidia/Cosmos3-Nano",
-            "serve_args": {
-                "cfg-parallel-size": 2,
-                "use-hsdp": true,
-                "hsdp-shard-size": 2,
-                "vae-patch-parallel-size": 2,
-                "vae-use-tiling": true,
-                "model-class-name": "Cosmos3OmniDiffusersPipeline",
-                "no-guardrails": true,
-                "stage-init-timeout": 900,
-                "init-timeout": 1200,
-                "enable-diffusion-pipeline-profiler": true
-            }
-        },
-        "benchmark_params": [
-            {
-                "name": "1280x720_frames189_steps4",
-                "dataset": "random",
-                "task": "v2v",
-                "num-prompts": 3,
-                "max-concurrency": 1,
-                "seed": 42,
-                "random-request-config": [
-                    {
-                        "width": 1280,
-                        "height": 720,
-                        "prompt": "Continue the same scene with smooth natural motion.",
-                        "num_inference_steps": 4,
-                        "num_frames": 189,
-                        "fps": 24,
-                        "weight": 1
-                    }
-                ],
-                "extra-body": {
-                    "guidance_scale": 6.0,
-                    "flow_shift": 10.0,
-                    "max_sequence_length": 4096,
-                    "guardrails": false,
-                    "condition_frame_indexes_vision": [
-                        0,
-                        1
-                    ],
-                    "condition_video_keep": "first"
-                },
-                "baseline": {
-                    "throughput_qps": 0.027,
-                    "latency_mean": 37.2117,
-                    "peak_memory_mb_mean": 30135.6
-                }
-            }
-        ]
-    }
-]

tests/dfx/perf/tests/test_hunyuan_image3_it2i.json

@@ -7,7 +7,6 @@
         "server_params": {
             "model": "tencent/HunyuanImage-3.0-Instruct",
             "serve_args": {
-                "trust-remote-code": true,
                 "deploy-config-inline": {
                     "type": "yaml",
                     "content": {

tests/dfx/reliability/invalid_param_test/test_invalid_image_editing.py

@@ -62,8 +62,8 @@ def _finalize_edits_form_data(template: dict[str, Any], server_model: str) -> di
         ),
         pytest.param(
             True,
-            {"prompt": "edit", "model": _IMAGE_EDITS_SERVER_MODEL, "layers": "1"},
-            ("Invalid layers", "2", "10"),
+            {"prompt": "edit", "model": _IMAGE_EDITS_SERVER_MODEL, "layers": "2"},
+            ("Invalid layers", "3", "1"),
             400,
             id="invalid_layers",
         ),

tests/dfx/reliability/invalid_param_test/test_invalid_image_generation.py

@@ -87,8 +87,8 @@ def _minimal_images_gen_json(omni_server: OmniServer) -> dict[str, object]:
         pytest.param(
             {"true_cfg_scale": 21.0}, ("true_cfg_scale", "less_than_equal", "20"), id="true_cfg_scale_above_max"
         ),
-        pytest.param({"layers": 1}, ("layers", "value_error", "2", "10"), id="layers_below_min"),
-        pytest.param({"layers": 11}, ("layers", "value_error", "2", "10"), id="layers_above_max"),
+        pytest.param({"layers": 2}, ("layers", "value_error", "3", "10"), id="layers_below_min"),
+        pytest.param({"layers": 11}, ("layers", "value_error", "3", "10"), id="layers_above_max"),
         pytest.param({"seed": -1}, ("seed", "greater_than_equal", "0"), id="seed_negative", marks=_SKIP_ISSUE_3649),
         pytest.param(
             {"seed": 2**32},