Fixed whisper example

jafioti · May 26, 2024 · ad2d73f · ad2d73f
1 parent 89b6d75
commit ad2d73f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/whisper/src/main.rs b/examples/whisper/src/main.rs
@@ -41,7 +41,7 @@ fn main() {
         .map(|_| (dec_cx.named_tensor("Keys"), dec_cx.named_tensor("Values")))
         .collect();
     cache_src.set_dyn(vec![], &[1, 6, 64, 0]);
-    let (logits, _, cache_dest) = decoder.forward((
+    let (logits, _, mut cache_dest) = decoder.forward((
         encoder_output,
         text_input,
         &cache_src,
@@ -85,6 +85,7 @@ fn main() {
             &mut text_input,
             &mut encoder_output,
             &mut cache_src,
+            &mut cache_dest,
             &mut logits,
             &mut decoder_params,
         ),