mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-05 04:04:28 +08:00
[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