From 0a2b5e87b699ff4b1f49ef2fce4905ea31c332a6 Mon Sep 17 00:00:00 2001 From: WeiQing Chen <40507679+david6666666@users.noreply.github.com> Date: Sun, 4 Jan 2026 19:56:50 +0800 Subject: [PATCH] [Misc] fix ring att log (#610) Signed-off-by: David Chen <530634352@qq.com> --- vllm_omni/diffusion/attention/parallel/ring.py | 2 +- vllm_omni/diffusion/data.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/vllm_omni/diffusion/attention/parallel/ring.py b/vllm_omni/diffusion/attention/parallel/ring.py index 784e28bd5..df145ac3b 100644 --- a/vllm_omni/diffusion/attention/parallel/ring.py +++ b/vllm_omni/diffusion/attention/parallel/ring.py @@ -120,7 +120,7 @@ class RingParallelAttention: if query.dtype == torch.float32 or not HAS_FLASH_ATTN: if not HAS_FLASH_ATTN and backend_pref != "sdpa": logger = init_logger(__name__) - logger.warning("Flash Attention is not available! Force enabling SDPA.") + logger.warning_once("Flash Attention is not available! Force enabling SDPA.") backend_pref = "sdpa" # Extract joint tensors diff --git a/vllm_omni/diffusion/data.py b/vllm_omni/diffusion/data.py index 6f54e9bd1..71cc4de3b 100644 --- a/vllm_omni/diffusion/data.py +++ b/vllm_omni/diffusion/data.py @@ -500,7 +500,7 @@ def get_cached_compilation_config(): def get_current_omni_diffusion_config() -> OmniDiffusionConfig: if _current_omni_diffusion_config is None: - logger.warning("Current OmniDiffusionConfig is not set.") + logger.warning_once("Current OmniDiffusionConfig is not set.") return OmniDiffusionConfig() return _current_omni_diffusion_config