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]skip to_tensor (#6342)

Browse Source
This commit is contained in:
sunxin
2026-02-04 17:25:19 +08:00
committed by GitHub
parent 26ba019e66
commit ef47e6eb46
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastdeploy/worker/input_batch.py
+3 -2
View File
@@ -751,8 +751,9 @@ def recover_batch_index_for_output(output_cls, index_to_batch_id, enable_pd_reor
is_not_swapped = all(i == v for i, v in index_to_batch_id.items()) or not enable_pd_reorder
# Create a new tensor to store the reordered results
sorted_keys = sorted(index_to_batch_id.keys())
index_to_batch_id_tmp = [index_to_batch_id[key] for key in sorted_keys]
index_to_batch_id_tensor = paddle.to_tensor(index_to_batch_id_tmp, dtype="int64")
if not is_not_swapped:
index_to_batch_id_tmp = [index_to_batch_id[key] for key in sorted_keys]
index_to_batch_id_tensor = paddle.to_tensor(index_to_batch_id_tmp, dtype="int64")
for recover_name in recover_list:
if isinstance(output_cls, dict):
recover_tensor = output_cls[recover_name]