From 8442edcd9b95971a45bce299cd284215c99d16ef Mon Sep 17 00:00:00 2001
From: Kaushik Ram Sadagopan <krs@fb.com>
Date: Tue, 17 Oct 2023 15:53:39 -0400
Subject: [PATCH] Fixing bug in create_default_sdpa() in wav2vec2 builder.
 (#105)

---
 src/fairseq2/models/wav2vec2/builder.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/fairseq2/models/wav2vec2/builder.py b/src/fairseq2/models/wav2vec2/builder.py
index a229ce608..a4e123317 100644
--- a/src/fairseq2/models/wav2vec2/builder.py
+++ b/src/fairseq2/models/wav2vec2/builder.py
@@ -365,7 +365,7 @@ def build_sdpa(self) -> SDPA:
                 dtype=self.dtype,
             )
 
-        return create_default_sdpa(self.config.attn_dropout_p)
+        return create_default_sdpa(attn_dropout_p=self.config.attn_dropout_p)
 
     def build_conformer_conv(self) -> ConformerConvolution:
         return ConformerConvolution(