[Perf][DreamZero] Remove redundant CUDA compile guard (#6983)

Signed-off-by: MikeyDong1 <xdong-1219@outlook.com>
This commit is contained in:
MikeyDong1 authored and GitHub committed 2026-09-04 09:01:05 -04:00
1 parent 10701e1838
commit 41f4f9d8c9
1 file changed
+1 -5
@@ -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,
)