From 198029a21f535cae0c92c415b2ffbfa343690a0a Mon Sep 17 00:00:00 2001 From: The praxis Authors Date: Fri, 29 Mar 2024 23:55:42 -0700 Subject: [PATCH] Cleans up a trivial unnecessary logic. PiperOrigin-RevId: 620433429 --- praxis/sample_decode.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/praxis/sample_decode.py b/praxis/sample_decode.py index c86f6c8e..67d16628 100644 --- a/praxis/sample_decode.py +++ b/praxis/sample_decode.py @@ -1281,10 +1281,6 @@ def put(target, key, val): result_callback.done_fn() if optimize_eos: - if fprop_for_prefix: - decode_length_shift = max_prefix_len - else: - decode_length_shift = 0 result = decoder_utils.collect_results_to_optimize_eos( result, decode_length_shift=max_prefix_len )