Fixing bug in create_default_sdpa() in wav2vec2 builder. (#105)

facebookresearch · Oct 17, 2023 · 8442edc · 8442edc
1 parent 4604d73
commit 8442edc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(