From 41f4f9d8c91a15b6d2f3b65516bfd885559d8e27 Mon Sep 17 00:00:00 2001 From: MikeyDong1 Date: Fri, 4 Sep 2026 06:01:05 -0700 Subject: [PATCH] [Perf][DreamZero] Remove redundant CUDA compile guard (#6983) Signed-off-by: MikeyDong1 --- vllm_omni/diffusion/models/dreamzero/pipeline_dreamzero.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/vllm_omni/diffusion/models/dreamzero/pipeline_dreamzero.py b/vllm_omni/diffusion/models/dreamzero/pipeline_dreamzero.py index 693766332..bf19f0a0f 100644 --- a/vllm_omni/diffusion/models/dreamzero/pipeline_dreamzero.py +++ b/vllm_omni/diffusion/models/dreamzero/pipeline_dreamzero.py @@ -1,5 +1,5 @@ # SPDX-License-Identifier: Apache-2.0 -# SPDX-FileCopyrightText: Copyright contributors to the vLLM project +# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project """DreamZero pipeline for vllm-omni. @@ -658,10 +658,6 @@ class DreamZeroPipeline(nn.Module, CFGParallelMixin): Wan ``feat_cache`` mutation is incompatible with CUDAGraph capture. DiT blocks use per-block ``fullgraph=True``. """ - if not torch.cuda.is_available(): - logger.info("DreamZero setup_compile skipped: CUDA not available.") - return - from vllm_omni.diffusion.models.dreamzero.wan_vae_feat_cache_patch import ( apply_wan_vae_feat_cache_tensor_patch, )