[Misc] fix ring att log (#610)
Signed-off-by: David Chen <530634352@qq.com>
This commit is contained in:
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
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
Reference in new issue
Block a user