Skip to content

Commit 0028a75

Browse files
HsChen-sysHaisheng Chen
authored andcommitted
Update ernie45_moe.py
define example_moe first Signed-off-by: Haisheng Chen <[email protected]>
1 parent 7ccbfb6 commit 0028a75

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

vllm/model_executor/models/ernie45_moe.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -642,6 +642,7 @@ def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""):
642642
self.num_expert_groups = 1
643643

644644
self.moe_layers: list[FusedMoE] = []
645+
example_moe = None
645646
for layer in self.model.layers:
646647
if isinstance(layer, PPMissingLayer):
647648
continue

0 commit comments

Comments
 (0)