from local

This commit is contained in:
dohe0342 2023-02-25 18:28:11 +09:00
parent 0b5398db0c
commit a29a52b500
3 changed files with 1 additions and 0 deletions

Binary file not shown.

View File

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