from local

This commit is contained in:
dohe0342 2023-02-25 18:31:39 +09:00
parent daa845d756
commit 9163475650
2 changed files with 0 additions and 1 deletions

Binary file not shown.

View File

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