Unverified Commit 819d548e authored by yihong's avatar yihong Committed by GitHub
Browse files

[BugFix] logger is not callable (#16312)


Signed-off-by: default avataryihong0618 <zouzou0208@gmail.com>
parent 477d2a8a
...@@ -149,8 +149,8 @@ class HPUAttentionImpl(AttentionImpl, torch.nn.Module): ...@@ -149,8 +149,8 @@ class HPUAttentionImpl(AttentionImpl, torch.nn.Module):
self.fused_scaled_dot_product_attention = ModuleFusedSDPA( self.fused_scaled_dot_product_attention = ModuleFusedSDPA(
FusedSDPA) FusedSDPA)
except ImportError: except ImportError:
logger().warning("Could not import HPU FusedSDPA kernel. " logger.warning("Could not import HPU FusedSDPA kernel. "
"vLLM will use native implementation.") "vLLM will use native implementation.")
suppored_head_sizes = HPUPagedAttention.get_supported_head_sizes() suppored_head_sizes = HPUPagedAttention.get_supported_head_sizes()
if head_size not in suppored_head_sizes: if head_size not in suppored_head_sizes:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment