from local

This commit is contained in:
dohe0342 2023-04-21 11:34:47 +09:00
parent 604e435c0f
commit ce3a5eac56
2 changed files with 4 additions and 1 deletions

View File

@ -358,7 +358,10 @@ def decode_one_batch(
)
else:
encoder_out, encoder_out_lens = model.encoder(x=feature, x_lens=feature_lens, prompt=model.prompt)
if 1:
print(encoder_out.size())
exit()
hyps = []
if params.decoding_method == "fast_beam_search":