Skip to content

Commit df40bed

Browse files
F1L1Pv2bobqianic
authored andcommitted
examples : Auto lowercase language parameter in main.cpp (ggml-org#1928)
* Auto lowercase language parameter * Update examples/main/main.cpp Co-authored-by: bobqianic <[email protected]> --------- Co-authored-by: bobqianic <[email protected]>
1 parent 3e5cd2d commit df40bed

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

examples/main/main.cpp

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,14 @@ struct whisper_params {
8282

8383
void whisper_print_usage(int argc, char ** argv, const whisper_params & params);
8484

85+
char* whisper_param_turn_lowercase(char* in){
86+
int string_len = strlen(in);
87+
for(int i = 0; i < string_len; i++){
88+
*(in+i) = tolower((unsigned char)*(in+i));
89+
}
90+
return in;
91+
}
92+
8593
bool whisper_params_parse(int argc, char ** argv, whisper_params & params) {
8694
for (int i = 1; i < argc; i++) {
8795
std::string arg = argv[i];
@@ -135,7 +143,7 @@ bool whisper_params_parse(int argc, char ** argv, whisper_params & params) {
135143
else if (arg == "-pc" || arg == "--print-colors") { params.print_colors = true; }
136144
else if (arg == "-pp" || arg == "--print-progress") { params.print_progress = true; }
137145
else if (arg == "-nt" || arg == "--no-timestamps") { params.no_timestamps = true; }
138-
else if (arg == "-l" || arg == "--language") { params.language = argv[++i]; }
146+
else if (arg == "-l" || arg == "--language") { params.language = whisper_param_turn_lowercase(argv[++i]); }
139147
else if (arg == "-dl" || arg == "--detect-language") { params.detect_language = true; }
140148
else if ( arg == "--prompt") { params.prompt = argv[++i]; }
141149
else if (arg == "-m" || arg == "--model") { params.model = argv[++i]; }

0 commit comments

Comments
 (0)