From eb3e7f2a4886fbb569d15427239a6eba8314cf60 Mon Sep 17 00:00:00 2001
From: Adam Kropp <30504149+akropp@users.noreply.github.com>
Date: Mon, 13 Nov 2023 13:47:17 -0500
Subject: [PATCH] Update src/mlpack/methods/ann/layer/multihead_attention.hpp

Co-authored-by: Ryan Curtin <ryan@ratml.org>
---
 src/mlpack/methods/ann/layer/multihead_attention.hpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/mlpack/methods/ann/layer/multihead_attention.hpp b/src/mlpack/methods/ann/layer/multihead_attention.hpp
index e9b407e322..7466517e67 100644
--- a/src/mlpack/methods/ann/layer/multihead_attention.hpp
+++ b/src/mlpack/methods/ann/layer/multihead_attention.hpp
@@ -192,7 +192,8 @@ class MultiheadAttentionType : public Layer<MatType>
 
   void ComputeOutputDimensions() override
   {
-    if (this->inputDimensions.size() < 2) {
+    if (this->inputDimensions.size() < 2)
+    {
       Log::Fatal << "Must have at least two dimensions for MultiHeadAttention: "
           << " [EmbeddingDim,SequenceLen]" << std::endl;
     }