Fix black style

This commit is contained in:
pkufool 2022-01-07 20:29:49 +08:00
parent 7dae4a11fd
commit 725f6ddb9b

View File

@ -294,7 +294,7 @@ def beam_search(
cached_key += f"-t-{t}" cached_key += f"-t-{t}"
if cached_key not in joint_cache: if cached_key not in joint_cache:
logits = model.joiner( logits = model.joiner(
current_encoder_out, decoder_out.unsqueeze(1) current_encoder_out, decoder_out.unsqueeze(1)
) )
# TODO(fangjun): Cache the blank posterior # TODO(fangjun): Cache the blank posterior