minor bug fixes for existing scripts

This commit is contained in:
jinzr 2023-08-11 20:51:36 +08:00
parent 91044348bb
commit 14f0cb5977
2 changed files with 2 additions and 2 deletions

View File

@ -274,7 +274,7 @@ def main():
params.update(vars(args))
token_table = k2.SymbolTable.from_file(params.tokens)
params.vocab_size = num_tokens(token_table)
params.vocab_size = num_tokens(token_table) + 1 # +1 for blank
params.blank_id = token_table["<blk>"]
assert params.blank_id == 0