[BugFix] fix error of import paddle.base.core.Config (#3761) (#3804)

* 延迟 import Config

* support chunked_prefill

* support chunked_prefill
This commit is contained in:
Yuanle Liu
2025-09-03 10:14:03 +08:00
committed by GitHub
parent 5cda326ba2
commit 174510180a
2 changed files with 20 additions and 3 deletions

View File

@@ -18,7 +18,6 @@ from abc import abstractmethod
import paddle
from paddle import nn
from paddle.base.core import Config
from paddleformers.utils.log import logger
try:
@@ -103,6 +102,8 @@ class DeepEPEngine:
self.deepep_engine = None
from paddle.base.core import Config
self.ep_config = Config(24, 6, 256)
self.num_max_dispatch_tokens_per_rank = num_max_dispatch_tokens_per_rank