mirror of
https://github.com/ggml-org/whisper.cpp.git
synced 2026-07-23 11:10:57 -05:00
0d14756929
* Add Whisper::Parakeet::Params * Add tests for Parakeet::Params * Remove unused variabel * Add callbacks to Parakeet::Params * Group callback and user_data params * Undefine local macros * Define GetParakeetParams * Remove unused variable * Use ITERATE_CALLBACK_PARAMS * Use ITERATE_CALLBACK_PARAMS instead of ITERATE_USER_DATA_PARAMS * Fix memsize * Remove unnecessary macros * Simplify params registration * Define Parakeet * Add hook methods to Parakeet::Params * Fix typo * Check callback container in GetParakeetParams * Reduce if * Free parakeet_full_params * Implement Parakeet::Context#initialize * Add TestParakeetContext * Add Parakeet::Segment * Prevent double-free * Add Parakeet::Context#transcribe * Add Parakeet::Context#each_segment * Define Parakeet::Segment attributes * Define Parakeet::Segment#deconstruct_keys * Add tests for Parakeet::Segment#deconstruct_keys * Run Parakeet::Context#transcribe without GVL * Make it to abort for Parakeet * Add Parakeet.log_set * Define Parakeet::Token * Define Parakeet::Segment#each_token * Implement some hooks of Parakeet::Params * Convert int to VALUE * Implement hooks for Parakeet * Implement Parakeet::Context#full * Add tests for Parakeet::Context#full * Add Parakeet to RBS * Fix ruby_whisper_parakeet_params_free * Free ruby_whisper_parakeet_context * Add tests for hooks * Add Parakeet section to README * Add more attributes of Parakeet::Context * Add tests for Parakeet::Context's attributes * Update RBS * Register parakeet-tdt-0.6b-v3 * Narrow scope of log constants * Extract activate and deactivate of log_queue * Make start_log_callback_thread private * Don't call start_log_callback_thread unncecessarilly * Early return from log_queue_enqueue when not active * Gropu log_queue members * is_active -> is_open * Fix English * Share parakeet full body function * ruby_whisper_parakeet_abort_callback_user_data -> ruby_whisper_abort_callback_user_data * NULL check for callback containers * Fix Parakeet.log_set * Omit Parakeet tests on CI * Extract Whisper::LogSettable * Join log callback thread in a log queue function * Revert Join log callback thread in a log queue function * Extract output methods to modules * Move Parakeet init functions into init_parakeet() * Add output methods to Parakeet classes * Add Parakeet's output methods to RBS * Use Whisper::Output in RBS * Add LogSettable to RBS * Fix module Token -> class Token * Add Parakeet::Model * Add test for Parakeet::Model * Add Parakeet::Model to RBS * Move position of Parakeet::Model in RBS * Parakeet -> TestBase::Parakeet * Add Parakeet::Context#model in RBS * Add Whisper::Output * Fix nil check * Define ruby_whisper_parakeet_model_memsize * Fix order of declaration in ruby_whisper_parakeet_model_get_xxx * Define Parakeet.system_info_str * Add test for Parakeet.system_info_str * Add signature of Parakeet.system_info_str * Define Parakeet::VERSION * Add test for Parakeet::VERSION * Add signature of Parakeet::VERSION * Add Parakeet::Context::Params * Make Parakeet::Context.new accept Context::Params * Add test for Parakeet::Context.new with Context::Params * Update RBS * Remove params from Parakeet::Params which are moved from whisper_parakeet_full_params * Remove tests for removed params * Make Parakeet tests follow original behavior changes * Add Parakeet model shortcuts * Alloc token data in factory instead of alloc func * Fix variable name * Update RBS * Refactor log settable module * Use log settable for Whisper * Address deadlock * Make test follow change of log queue implementation * Refactor to make abort callback use the same way to parakeet's way * Remove redundant structs * Fix test name * Fix README * Add missing parallel transcription * Fix test for parakeet info * Remove removed params * Wait for logs dequeued * Fix instance variable name * Load etc feature * Remove unnecessary comment * Remove unnecessary thread safety check * Remove outdated comment * Skip downloading model if cache exists * Change Hugging Face URI for Parakeet models * Bump required Ruby version to 3.3 * Fix English
48 lines
1.7 KiB
C
48 lines
1.7 KiB
C
#ifndef RUBY_WHISPER_LOG_SETTABLE_H
|
|
#define RUBY_WHISPER_LOG_SETTABLE_H
|
|
|
|
#define LOG_SETTABLE_SETUP(log_queue, mod, log_set) \
|
|
static VALUE \
|
|
ruby_whisper_##log_queue##_s_drain_logs(VALUE self) \
|
|
{ \
|
|
return ruby_whisper_log_queue_drain(&log_queue); \
|
|
} \
|
|
static void \
|
|
ruby_whisper_##log_queue##_log_callback(enum ggml_log_level level, const char *text, void *user_data) \
|
|
{ \
|
|
ruby_whisper_log_queue_enqueue(&log_queue, level, text); \
|
|
} \
|
|
static VALUE \
|
|
ruby_whisper_##log_queue##_s_log_set(VALUE self, VALUE log_callback, VALUE user_data) \
|
|
{ \
|
|
rb_iv_set(self, "@log_callback", log_callback); \
|
|
rb_iv_set(self, "@log_callback_user_data", user_data); \
|
|
if (NIL_P(log_callback)) { \
|
|
log_set(NULL, NULL); \
|
|
} else { \
|
|
ruby_whisper_log_queue_open(&log_queue); \
|
|
rb_funcall((mod), id_start_log_callback_thread, 0); \
|
|
log_set(ruby_whisper_##log_queue##_log_callback, NULL); \
|
|
} \
|
|
return Qnil; \
|
|
} \
|
|
static void \
|
|
ruby_whisper_##log_queue##_end_proc(VALUE args) \
|
|
{ \
|
|
ruby_whisper_log_queue_close(&log_queue); \
|
|
VALUE log_callback_thread = rb_ivar_get(mod, id_log_callback_thread); \
|
|
if (!NIL_P(log_callback_thread) && RTEST(rb_funcall(log_callback_thread, id_alive_p, 0))) { \
|
|
rb_funcall(log_callback_thread, id_join, 0); \
|
|
} \
|
|
}
|
|
|
|
#define LOG_SETTABLE_INIT(log_queue, mod) \
|
|
ruby_whisper_log_queue_initialize(&log_queue); \
|
|
rb_define_singleton_method(mod, "drain_logs", ruby_whisper_##log_queue##_s_drain_logs, 0); \
|
|
rb_define_singleton_method(mod, "log_set", ruby_whisper_##log_queue##_s_log_set, 2); \
|
|
rb_set_end_proc(ruby_whisper_##log_queue##_end_proc, Qnil); \
|
|
rb_extend_object(mod, mLogSettable); \
|
|
rb_funcall(mLogSettable, id_extended, 1, mod);
|
|
|
|
#endif
|