Skip to content

Commit 153bc70

Browse files
committed
fix(eagle3):change pytorch_weights_path to speculative_model_dir
1 parent 9fe49f1 commit 153bc70

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tests/integration/defs/accuracy/references/mmlu.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ Qwen3/Qwen3-8B:
143143
- quant_algo: FP8_BLOCK_SCALES
144144
accuracy: 76.12
145145
- accuracy: 76.12
146-
- spec_dec_algo: EAGLE3
146+
- spec_dec_algo: Eagle
147147
accuracy: 76.12
148148
Qwen3/Qwen3-30B-A3B:
149149
- quant_algo: FP8_BLOCK_SCALES

tests/integration/defs/accuracy/test_llm_api_pytorch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1661,8 +1661,8 @@ def test_eagle3(self):
16611661

16621662
draft_len = 4
16631663
spec_config = EagleDecodingConfig(max_draft_len=draft_len,
1664-
pytorch_weights_path=eagle_model_dir,
1665-
eagle3_one_model=False)
1664+
speculative_model_dir=eagle_model_dir
1665+
)
16661666

16671667
llm = LLM(model=target_model_dir,
16681668
**pytorch_config,

0 commit comments

Comments
 (0)