Support c-api (#317)

This commit is contained in:
Wei Kang
2023-09-18 16:24:57 +08:00
committed by GitHub
parent bd18e073f4
commit a5d1c90807
5 changed files with 61 additions and 7 deletions

View File

@@ -52,7 +52,21 @@ static struct cag_option options[] = {
.access_name = "decoding-method",
.value_name = "decoding-method",
.description =
"Decoding method: greedy_search (default), modified_beam_search"}};
"Decoding method: greedy_search (default), modified_beam_search"},
{.identifier = 'f',
.access_letters = NULL,
.access_name = "hotwords-file",
.value_name = "hotwords-file",
.description = "The file containing hotwords, one words/phrases per line, "
"and for each phrase the bpe/cjkchar are separated by a "
"space. For example: ▁HE LL O ▁WORLD, 你 好 世 界"},
{.identifier = 's',
.access_letters = NULL,
.access_name = "hotwords-score",
.value_name = "hotwords-score",
.description = "The bonus score for each token in hotwords. Used only "
"when decoding_method is modified_beam_search"},
};
const char *kUsage =
"\n"
@@ -130,6 +144,12 @@ int32_t main(int32_t argc, char *argv[]) {
case 'm':
config.decoding_method = value;
break;
case 'f':
config.hotwords_file = value;
break;
case 's':
config.hotwords_score = atof(value);
break;
case 'h': {
fprintf(stderr, "%s\n", kUsage);
exit(0);