tweak sampling strat for whisper to hopefully decrease hallucination (#109)
This commit is contained in:
parent
7be3fac4b1
commit
cdfe15c898
1 changed files with 4 additions and 1 deletions
|
|
@ -441,7 +441,10 @@ impl TranscriptionManager {
|
|||
let settings = get_settings(&self.app_handle);
|
||||
|
||||
// Initialize parameters
|
||||
let mut params = FullParams::new(SamplingStrategy::default());
|
||||
let mut params = FullParams::new(SamplingStrategy::BeamSearch {
|
||||
beam_size: 3,
|
||||
patience: -1.0,
|
||||
});
|
||||
let language = Some(settings.selected_language.as_str());
|
||||
params.set_language(language);
|
||||
params.set_print_special(false);
|
||||
|
|
|
|||
Loading…
Reference in a new issue