mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-12-24 13:28:13 +08:00
default enable chunked prefill (#3731)
* add error traceback info * update error msg * update code * default enable chunked prefill * update code * update code * add envs * update code --------- Co-authored-by: Jiang-Jia-Jun <163579578+Jiang-Jia-Jun@users.noreply.github.com>
This commit is contained in:
@@ -1233,6 +1233,15 @@ class FDConfig:
|
||||
|
||||
self.paddle_commit_id = paddle.version.commit
|
||||
|
||||
if self.cache_config.enable_chunked_prefill:
|
||||
self.force_chunked_prefill = int(envs.FD_FORCE_CHUNKED_PREFILL)
|
||||
if (
|
||||
self.speculative_config is not None
|
||||
and self.speculative_config.method in ["mtp"]
|
||||
and not self.force_chunked_prefill
|
||||
):
|
||||
self.cache_config.enable_chunked_prefill = False
|
||||
|
||||
if self.max_num_batched_tokens is None:
|
||||
if self.cache_config.enable_chunked_prefill:
|
||||
self.max_num_batched_tokens = 2048
|
||||
|
||||
Reference in New Issue
Block a user