diff --git a/whisper-cpp-2023/rust.d/Cargo.lock b/whisper-cpp-2023/rust.d/Cargo.lock index dbb80de..52f9276 100644 --- a/whisper-cpp-2023/rust.d/Cargo.lock +++ b/whisper-cpp-2023/rust.d/Cargo.lock @@ -190,6 +190,15 @@ version = "1.0.1" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "b9b1a3d5f46d53f4a3478e2be4a5a5ce5108ea58b100dcd139830eae7f79a3a1" +[[package]] +name = "rust-whisper" +version = "0.1.0" +dependencies = [ + "tokio", + "wav", + "whisper-rs", +] + [[package]] name = "rustc-hash" version = "1.1.0" @@ -250,15 +259,6 @@ dependencies = [ "once_cell", ] -[[package]] -name = "whisper-cpp-er" -version = "0.1.0" -dependencies = [ - "tokio", - "wav", - "whisper-rs", -] - [[package]] name = "whisper-rs" version = "0.5.0" diff --git a/whisper-cpp-2023/rust.d/src/main.rs b/whisper-cpp-2023/rust.d/src/main.rs index ca0ae67..bb0d57d 100644 --- a/whisper-cpp-2023/rust.d/src/main.rs +++ b/whisper-cpp-2023/rust.d/src/main.rs @@ -7,7 +7,9 @@ fn main() { // create a params object let mut params = FullParams::new(SamplingStrategy::Greedy { best_of: 0 }); - params.set_n_threads(4); + params.set_n_threads( + std::env::var("P").unwrap_or(String::from("1")).parse::().expect("$P must be a number") + ); params.set_translate(false); params.set_language(Some("en")); params.set_print_special(false);