[Misc] fix ring att log (#610)

Signed-off-by: David Chen <530634352@qq.com>
This commit is contained in:
WeiQing Chen authored and GitHub committed 2026-01-04 19:56:50 +08:00
1 parent 3c2661da16
commit 0a2b5e87b6
2 files changed
+2 -2

No files matched your search

@@ -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
+1 -1
View File
@@ -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