fix wav2vec2 compilation

OpenNMT · Sep 18, 2024 · 2d3dad5 · 2d3dad5
1 parent df5491d
commit 2d3dad5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/layers/flash_attention.cc b/src/layers/flash_attention.cc
@@ -18,6 +18,7 @@ namespace ctranslate2 {
     void FlashMultiHeadAttention::operator()(const StorageView& queries,
                                              const StorageView&,
                                              const StorageView* values_lengths,
+                                             const StorageView* values_offsets,
                                              StorageView& output,
                                              StorageView* cached_keys,
                                              StorageView* cached_values,

diff --git a/src/layers/wav2vec2bert.cc b/src/layers/wav2vec2bert.cc
@@ -51,6 +51,7 @@ namespace ctranslate2 {
       (*_self_attention)(buffer1,
                          buffer1,
                          nullptr,
+                         nullptr,
                          buffer2,
                          nullptr,
                          nullptr,
@@ -147,6 +148,7 @@ namespace ctranslate2 {
       (*_self_attention)(buffer2,
                          buffer2,
                          nullptr,
+                         nullptr,
                          buffer1,
                          nullptr,
                          nullptr,