apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 00:17:25 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

[Others] remove stop_nums (#6182)

Browse Source
This commit is contained in:
周周周
2026-01-26 12:12:47 +08:00
committed by GitHub
parent 84a1780814
commit 0966df78dc
13 changed files with 14 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastdeploy/model_executor/pre_and_post_process.py
-4
View File
@@ -397,7 +397,6 @@ def post_process_normal(
sampler_output.sampled_token_ids,
model_output.input_ids,
share_inputs["block_tables"],
model_output.stop_nums,
model_output.next_tokens,
model_output.is_block_step,
block_size,
@@ -410,7 +409,6 @@ def post_process_normal(
model_output.seq_lens_encoder,
model_output.seq_lens_decoder,
model_output.input_ids,
model_output.stop_nums,
sampler_output.sampled_token_ids,
model_output.is_block_step,
)
@@ -497,7 +495,6 @@ def post_process_specualate(
model_output.stop_flags,
model_output.seq_lens_this_time,
model_output.is_block_step,
model_output.stop_nums,
model_output.mask_rollback,
)
@@ -930,7 +927,6 @@ def post_process_pooling(
dummy_sampled_tokens,
model_output.input_ids,
share_inputs["block_tables"],
model_output.stop_nums,
model_output.next_tokens,
model_output.is_block_step,
block_size,