From 7ba4edc3725f6595c55dbe71537a2063faa83b01 Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Mon, 21 Aug 2023 16:53:36 -0700 Subject: [PATCH] Internal Change PiperOrigin-RevId: 558937644 --- .../cc/components/processors/proto/transformer_params.proto | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mediapipe/tasks/cc/components/processors/proto/transformer_params.proto b/mediapipe/tasks/cc/components/processors/proto/transformer_params.proto index 8c1daf277..b2d13c3a2 100644 --- a/mediapipe/tasks/cc/components/processors/proto/transformer_params.proto +++ b/mediapipe/tasks/cc/components/processors/proto/transformer_params.proto @@ -43,4 +43,7 @@ message TransformerParameters { // Number of stacked transformers, `N` in the paper. int32 num_stacks = 7; + + // Whether to use Multi-Query-Attention (MQA). + bool use_mqa = 8; }