gogs
/
rust-whisper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

$P to control rust threads

master
Bel LaPointe 2023-03-31 10:39:25 -06:00
parent 634d182afe
commit 8fb600b24e
2 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
whisper-cpp-2023/rust.d/Cargo.lock generated
View File

@ -190,6 +190,15 @@ version = "1.0.1"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b9b1a3d5f46d53f4a3478e2be4a5a5ce5108ea58b100dcd139830eae7f79a3a1" checksum = "b9b1a3d5f46d53f4a3478e2be4a5a5ce5108ea58b100dcd139830eae7f79a3a1"
[[package]]
name = "rust-whisper"
version = "0.1.0"
dependencies = [
"tokio",
"wav",
"whisper-rs",
]
[[package]] [[package]]
name = "rustc-hash" name = "rustc-hash"
version = "1.1.0" version = "1.1.0"
@ -250,15 +259,6 @@ dependencies = [
"once_cell", "once_cell",
] ]
[[package]]
name = "whisper-cpp-er"
version = "0.1.0"
dependencies = [
"tokio",
"wav",
"whisper-rs",
]
[[package]] [[package]]
name = "whisper-rs" name = "whisper-rs"
version = "0.5.0" version = "0.5.0"

4
whisper-cpp-2023/rust.d/src/main.rs
View File

@ -7,7 +7,9 @@ fn main() {
// create a params object // create a params object
let mut params = FullParams::new(SamplingStrategy::Greedy { best_of: 0 }); 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::<i32>().expect("$P must be a number")
);
params.set_translate(false); params.set_translate(false);
params.set_language(Some("en")); params.set_language(Some("en"));
params.set_print_special(false); params.set_print_special(false);