from local

This commit is contained in:
dohe0342 2023-02-25 16:13:19 +09:00
parent ca7bea83e7
commit 7d5afcce46
2 changed files with 0 additions and 1 deletions

View File

@ -423,7 +423,6 @@ def decode_one_batch(
token_ids=unsorted_token_ids, token_ids=unsorted_token_ids,
sos_id=graph_compiler.sos_id, sos_id=graph_compiler.sos_id,
eos_id=graph_compiler.eos_id, eos_id=graph_compiler.eos_id,
warmup=warmup,
return_output=True, return_output=True,
) )
print(pred.size()) print(pred.size())