Skip to content

Commit

Permalink
Fix warning
Browse files Browse the repository at this point in the history
  • Loading branch information
slyalin committed Nov 14, 2024
1 parent 6452b3d commit 033ab9d
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -155,7 +155,7 @@ ov::OutputVector group_query_attention(const ov::frontend::onnx::Node& node) {
const auto num_heads = node.get_attribute_value<int64_t>("num_heads");

if(ov::op::util::is_null(nodes[1]) || ov::op::util::is_null(nodes[2])) {
const auto split_result = detail::split_to_QKV(input, num_heads, {});
const auto split_result = detail::split_to_QKV(input, static_cast<int>(num_heads), {});
Q = split_result[0];
K = split_result[1];
V = split_result[2];
Expand Down

0 comments on commit 033ab9d

Please sign in to comment.