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

ep support logprob (#4089) (#4151)

Browse Source
This commit is contained in:
chen
2025-09-19 14:07:31 +08:00
committed by GitHub
parent a685e5ad35
commit 66a98b44ed
3 changed files with 11 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
fastdeploy/engine/args_utils.py
View File

@@ -401,8 +401,6 @@ class EngineArgs:
if self.enable_logprob:
if self.speculative_config is not None:
raise NotImplementedError("Logprob does not support speculation_config.")
if self.enable_expert_parallel:
raise NotImplementedError("Logprob does not support enable_expert_parallel.")
if not current_platform.is_cuda():
raise NotImplementedError("Only CUDA platform supports logprob.")
if self.speculative_config is not None: