apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 00:33:03 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

enable dcu ci (#3402)

Browse Source
This commit is contained in:
lifulll
2025-08-29 10:23:08 +08:00
committed by GitHub
parent 73d60fe64d
commit 72094d4d82
11 changed files with 295 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
fastdeploy/model_executor/pre_and_post_process.py
View File

@@ -218,7 +218,7 @@ def post_process_normal(
model_output.stop_flags,
)
if current_platform.is_cuda() or current_platform.is_iluvatar():
if current_platform.is_cuda() or current_platform.is_iluvatar() or current_platform.is_dcu():
set_stop_value_multi_ends(
sampler_output.sampled_token_ids,
model_output.stop_flags,