Skip to content

Commit 1cc273f

Browse files
committed
Refactor whisper parameter type casting for improved clarity and consistency
1 parent 9cd26c4 commit 1cc273f

File tree

1 file changed

+19
-19
lines changed

1 file changed

+19
-19
lines changed

src/whisper-utils/whisper-params.cpp

Lines changed: 19 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -103,10 +103,10 @@ void apply_whisper_params_from_settings(whisper_full_params &params, obs_data_t
103103
{
104104
params = whisper_full_default_params(
105105
(whisper_sampling_strategy)obs_data_get_int(settings, "strategy"));
106-
params.n_threads = obs_data_get_int(settings, "n_threads");
107-
params.n_max_text_ctx = obs_data_get_int(settings, "n_max_text_ctx");
108-
params.offset_ms = obs_data_get_int(settings, "offset_ms");
109-
params.duration_ms = obs_data_get_int(settings, "duration_ms");
106+
params.n_threads = (int)obs_data_get_int(settings, "n_threads");
107+
params.n_max_text_ctx = (int)obs_data_get_int(settings, "n_max_text_ctx");
108+
params.offset_ms = (int)obs_data_get_int(settings, "offset_ms");
109+
params.duration_ms = (int)obs_data_get_int(settings, "duration_ms");
110110
params.translate = obs_data_get_bool(settings, "whisper_translate");
111111
params.no_context = obs_data_get_bool(settings, "no_context");
112112
params.no_timestamps = obs_data_get_bool(settings, "no_timestamps");
@@ -116,13 +116,13 @@ void apply_whisper_params_from_settings(whisper_full_params &params, obs_data_t
116116
params.print_realtime = obs_data_get_bool(settings, "print_realtime");
117117
params.print_timestamps = obs_data_get_bool(settings, "print_timestamps");
118118
params.token_timestamps = obs_data_get_bool(settings, "token_timestamps");
119-
params.thold_pt = obs_data_get_double(settings, "thold_pt");
120-
params.thold_ptsum = obs_data_get_double(settings, "thold_ptsum");
121-
params.max_len = obs_data_get_int(settings, "max_len");
119+
params.thold_pt = (float)obs_data_get_double(settings, "thold_pt");
120+
params.thold_ptsum = (float)obs_data_get_double(settings, "thold_ptsum");
121+
params.max_len = (int)obs_data_get_int(settings, "max_len");
122122
params.split_on_word = obs_data_get_bool(settings, "split_on_word");
123-
params.max_tokens = obs_data_get_int(settings, "max_tokens");
123+
params.max_tokens = (int)obs_data_get_int(settings, "max_tokens");
124124
params.debug_mode = obs_data_get_bool(settings, "debug_mode");
125-
params.audio_ctx = obs_data_get_int(settings, "audio_ctx");
125+
params.audio_ctx = (int)obs_data_get_int(settings, "audio_ctx");
126126
params.tdrz_enable = obs_data_get_bool(settings, "tdrz_enable");
127127
params.suppress_regex = obs_data_get_string(settings, "suppress_regex");
128128
params.initial_prompt = obs_data_get_string(settings, "initial_prompt");
@@ -131,16 +131,16 @@ void apply_whisper_params_from_settings(whisper_full_params &params, obs_data_t
131131
params.suppress_blank = obs_data_get_bool(settings, "suppress_blank");
132132
params.suppress_non_speech_tokens =
133133
obs_data_get_bool(settings, "suppress_non_speech_tokens");
134-
params.temperature = obs_data_get_double(settings, "temperature");
135-
params.max_initial_ts = obs_data_get_double(settings, "max_initial_ts");
136-
params.length_penalty = obs_data_get_double(settings, "length_penalty");
137-
params.temperature_inc = obs_data_get_double(settings, "temperature_inc");
138-
params.entropy_thold = obs_data_get_double(settings, "entropy_thold");
139-
params.logprob_thold = obs_data_get_double(settings, "logprob_thold");
140-
params.no_speech_thold = obs_data_get_double(settings, "no_speech_thold");
141-
params.greedy.best_of = obs_data_get_int(settings, "greedy.best_of");
142-
params.beam_search.beam_size = obs_data_get_int(settings, "beam_search.beam_size");
143-
params.beam_search.patience = obs_data_get_double(settings, "beam_search.patience");
134+
params.temperature = (float)obs_data_get_double(settings, "temperature");
135+
params.max_initial_ts = (float)obs_data_get_double(settings, "max_initial_ts");
136+
params.length_penalty = (float)obs_data_get_double(settings, "length_penalty");
137+
params.temperature_inc = (float)obs_data_get_double(settings, "temperature_inc");
138+
params.entropy_thold = (float)obs_data_get_double(settings, "entropy_thold");
139+
params.logprob_thold = (float)obs_data_get_double(settings, "logprob_thold");
140+
params.no_speech_thold = (float)obs_data_get_double(settings, "no_speech_thold");
141+
params.greedy.best_of = (int)obs_data_get_int(settings, "greedy.best_of");
142+
params.beam_search.beam_size = (int)obs_data_get_int(settings, "beam_search.beam_size");
143+
params.beam_search.patience = (float)obs_data_get_double(settings, "beam_search.patience");
144144
}
145145

146146
void add_whisper_params_group_properties(obs_properties_t *ppts)

0 commit comments

Comments
 (0)