From 517bb0b27fc7a9086e3a7a8317e4328000faeb75 Mon Sep 17 00:00:00 2001 From: Adam Kropp <30504149+akropp@users.noreply.github.com> Date: Wed, 8 Nov 2023 07:55:13 -0500 Subject: [PATCH] Update src/mlpack/methods/ann/layer/multihead_attention.hpp Co-authored-by: Ryan Curtin --- src/mlpack/methods/ann/layer/multihead_attention.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/mlpack/methods/ann/layer/multihead_attention.hpp b/src/mlpack/methods/ann/layer/multihead_attention.hpp index fa665ca0d1..a6f294dad5 100644 --- a/src/mlpack/methods/ann/layer/multihead_attention.hpp +++ b/src/mlpack/methods/ann/layer/multihead_attention.hpp @@ -180,8 +180,8 @@ class MultiheadAttentionType : public Layer srcSeqLen = this->inputDimensions[1]; if (embedDim % numHeads != 0) { - Log::Fatal << "Embedding dimension must be divisible by number of \ - attention heads." << std::endl; + Log::Fatal << "Embedding dimension must be divisible by number of " + << "attention heads." << std::endl; } headDim = embedDim / numHeads;