Fix CI errors. (#310)

This commit is contained in:
Fangjun Kuang 2022-04-12 09:09:56 +08:00 committed by GitHub
parent 118e195004
commit bdeff338c2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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