[BugFix] logger is not callable (#16312)
Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
parent
477d2a8aa2
commit
819d548e8a
@ -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:
|
||||||
|
Loading…
x
Reference in New Issue
Block a user