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

[Model Runner] Deprecate not_need_stop (#6356)

Browse Source 
* Deprecate not_need_stop
This commit is contained in:
sunxin
2026-03-05 10:55:42 +08:00
committed by GitHub
parent fa4815b93a
commit 0dc7034ce0
12 changed files with 534 additions and 165 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastdeploy/worker/input_batch.py
+1 -1
View File
@@ -152,7 +152,7 @@ class InputBatch:
self.seq_lens_this_time_cpu = paddle.full([max_num_seqs, 1], 0, dtype="int32", device="cpu")
self.is_block_step_cpu = paddle.full([max_num_seqs], False, dtype="bool", device="cpu")
else:
self.not_need_stop = paddle.full([1], False, dtype="bool").pin_memory()
self.not_need_stop = paddle.full([1], False, dtype="bool").cpu()
self.sampled_token_ids = paddle.full([max_num_seqs, 1], -1, dtype="int64").pin_memory()
self.seq_lens_this_time_cpu = paddle.full([max_num_seqs, 1], 0, dtype="int32").pin_memory()
self.is_block_step_cpu = paddle.full([max_num_seqs], False, dtype="bool").pin_memory()