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