We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a25e2e4 commit 64ecec9Copy full SHA for 64ecec9
1 file changed
machine/translation/huggingface/hugging_face_nmt_engine.py
@@ -164,9 +164,8 @@ def _try_translate_n_batch(
164
builder = TranslationResultBuilder(input_tokens)
165
for token, score in zip(output["translation_tokens"], output["token_scores"]):
166
builder.append_token(token, TranslationSources.NMT, exp(score))
167
- builder.set_sequence_confidence(
168
- exp(output["sequence_score"]) if output["sequence_score"] is not None else -1
169
- )
+ if output["sequence_score"] is not None:
+ builder.set_sequence_confidence(exp(output["sequence_score"]))
170
word_pairs: Optional[Collection[Union[AlignedWordPair, Tuple[int, int]]]] = None
171
if output.get("token_attentions") is not None:
172
src_indices = torch.argmax(output["token_attentions"], dim=1).tolist()
0 commit comments