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

Revert "try rollback to whisper-rs5 to avoid gpu in whispercpp but no luck" 

This reverts commit a483aaf25c.
master
Bel LaPointe 2023-11-08 11:35:31 -07:00
parent a483aaf25c
commit 50058037eb
3 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
rust-whisper.d/Cargo.lock generated
View File

@ -261,20 +261,21 @@ dependencies = [
[[package]] [[package]]
name = "whisper-rs" name = "whisper-rs"
version = "0.5.0" version = "0.8.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "fa7e1b9b003aa3285a0e4469219566266aa1d51ced1be38587251a4f713a1677" checksum = "2c950fb18ad556b053ba615b88fd4d01ed6020be740c3371eb0fc4aec64a0639"
dependencies = [ dependencies = [
"whisper-rs-sys", "whisper-rs-sys",
] ]
[[package]] [[package]]
name = "whisper-rs-sys" name = "whisper-rs-sys"
version = "0.3.1" version = "0.6.1"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "97a389dc665c7354ba6b1982850d4ba05b862907e535708ebdec92cbd9c599e8" checksum = "094a5bd86f6f52562bbc74c28f27cd80197e54656cfb7213cf4ba37b5246cc9e"
dependencies = [ dependencies = [
"bindgen", "bindgen",
"cfg-if",
] ]
[[package]] [[package]]

2
rust-whisper.d/Cargo.toml
View File

@ -6,6 +6,6 @@ edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies] [dependencies]
whisper-rs = "0.5.0" whisper-rs = "0.8.0"
wav = "1" wav = "1"
tokio = "1.27" tokio = "1.27"

10
rust-whisper.d/src/main.rs
View File

@ -1,10 +1,10 @@
use whisper_rs::{WhisperContext, FullParams, SamplingStrategy}; use whisper_rs::{WhisperContext, FullParams, SamplingStrategy};
fn main() { fn main() {
let mut ctx = WhisperContext::new( let ctx = WhisperContext::new(
&std::env::var("MODEL").unwrap_or(String::from("../models/ggml-tiny.en.bin")) &std::env::var("MODEL").unwrap_or(String::from("../models/ggml-tiny.en.bin"))
).expect("failed to load model"); ).expect("failed to load model");
//let mut state = ctx.create_state().expect("failed to create state"); let mut state = ctx.create_state().expect("failed to create state");
// 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 });
@ -26,11 +26,11 @@ fn main() {
let data16 = data.as_sixteen().expect("wav is not 32bit floats"); let data16 = data.as_sixteen().expect("wav is not 32bit floats");
let audio_data = &whisper_rs::convert_integer_to_float_audio(&data16); let audio_data = &whisper_rs::convert_integer_to_float_audio(&data16);
ctx.full(params, &audio_data[..]).expect("failed to run model"); state.full(params, &audio_data[..]).expect("failed to run model");
let num_segments = ctx.full_n_segments(); //.expect("failed to get number of segments"); let num_segments = state.full_n_segments().expect("failed to get number of segments");
for i in 0..num_segments { for i in 0..num_segments {
let segment = ctx.full_get_segment_text(i).expect("failed to get segment"); let segment = state.full_get_segment_text(i).expect("failed to get segment");
print!("{} ", segment); print!("{} ", segment);
} }
println!(""); println!("");