Fix CI errors.

This commit is contained in:
Fangjun Kuang 2022-04-12 08:50:35 +08:00
parent 118e195004
commit 73c0f2aacd

View File

@ -174,6 +174,7 @@ def main():
# <blk> is defined in local/train_bpe_model.py
params.blank_id = sp.piece_to_id("<blk>")
params.unk_id = sp.piece_to_id("<unk>")
params.vocab_size = sp.get_piece_size()
logging.info(f"{params}")