Fixed weight loading

jafioti · May 19, 2024 · c11fc64 · c11fc64
1 parent 706758b
commit c11fc64
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/examples/whisper/src/main.rs b/examples/whisper/src/main.rs
@@ -108,14 +108,14 @@ fn main() {
     print!("Loading weights");
     std::io::stdout().flush().unwrap();
     let now = std::time::Instant::now();
-    audio_input.set_dyn(vec![], &[1, 80, 0]);
-    enc_cx.set_dyn_dim('d', 0);
+    audio_input.set_dyn(vec![0.; 160], &[1, 80, 2]);
+    enc_cx.set_dyn_dim('d', 1);
     enc_cx.execute();
     delete_inputs(downstream(encoder_params, &enc_cx), &mut enc_cx);
-    text_input.set_dyn(vec![], &[1, 0]);
-    dec_cx.set_dyn_dim('e', 0);
+    text_input.set_dyn(vec![0.], &[1, 1]);
+    dec_cx.set_dyn_dim('e', 1);
     dec_cx.set_dyn_dim('p', 0);
-    dec_cx.set_dyn_dim('t', 0);
+    dec_cx.set_dyn_dim('t', 1);
     transfer_data(encoded, &mut enc_cx, &encoder_output, &mut dec_cx);
     dec_cx.execute();
     logits.drop();