[Perf][DreamZero] Remove redundant CUDA compile guard (#6983)
Signed-off-by: MikeyDong1 <xdong-1219@outlook.com>
This commit is contained in:
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,
|
||||
)
|
||||
|
||||
Reference in new issue
Block a user