Skip to content

Commit

Permalink
rcppexports add flash attention
Browse files Browse the repository at this point in the history
  • Loading branch information
jwijffels committed Oct 7, 2024
1 parent 575bf74 commit 0046ba7
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions R/RcppExports.R
Original file line number Diff line number Diff line change
@@ -1,6 +1,18 @@
# Generated by using Rcpp::compileAttributes() -> do not edit by hand
# Generator token: 10BE3573-1514-4C36-9D1C-5A225CD40393

whisper_load_model <- function(model, use_gpu = FALSE, flash_attn = FALSE) {
.Call('_audio_whisper_whisper_load_model', PACKAGE = 'audio.whisper', model, use_gpu, flash_attn)
}

whisper_encode <- function(model, path, language, token_timestamps = FALSE, translate = FALSE, duration = 0L, offset = 0L, trace = 1L, n_threads = 1L, n_processors = 1L, entropy_thold = 2.40, logprob_thold = -1.00, beam_size = -1L, best_of = 5L, split_on_word = FALSE, max_context = -1L, prompt = "", print_special = FALSE, diarize = FALSE, diarize_percent = 1.1) {
.Call('_audio_whisper_whisper_encode', PACKAGE = 'audio.whisper', model, path, language, token_timestamps, translate, duration, offset, trace, n_threads, n_processors, entropy_thold, logprob_thold, beam_size, best_of, split_on_word, max_context, prompt, print_special, diarize, diarize_percent)
}

whisper_print_benchmark <- function(model, n_threads = 1L) {
invisible(.Call('_audio_whisper_whisper_print_benchmark', PACKAGE = 'audio.whisper', model, n_threads))
}

whisper_language_info <- function() {
.Call('_audio_whisper_whisper_language_info', PACKAGE = 'audio.whisper')
}
Expand Down

0 comments on commit 0046ba7

Please sign in to comment.