from local

This commit is contained in:
dohe0342 2023-02-25 18:29:02 +09:00
parent a29a52b500
commit 6dc64702db
2 changed files with 1 additions and 1 deletions

Binary file not shown.

View File

@ -954,7 +954,7 @@ def rescore_with_attention_decoder(
assert nll.shape[0] == len(token_ids)
attention_scores = -nll.sum(dim=1)
attention_scores.size())
attention_scores.size()
if ngram_lm_scale is None:
ngram_lm_scale_list = [0.]