[Iluvatar][CI] Fix cannot import get_stop (#6165)

2026-04-23 00:17:25 +08:00 · 2026-02-10 16:57:23 +08:00
parent d937d6ebfd
commit 60e75ea8e8
7 changed files with 122 additions and 6 deletions
@@ -882,6 +882,7 @@ def rebuild_padding(
            seq_lens_decoder,
            seq_lens_encoder,
            batch_id_per_token_output,
+            cu_seqlens_q_output,
            first_token_out,
            enable_logprob,
        )