Skip to content

Commit 2dfca2e

Browse files
author
krizaltang
committed
[fix]: Fix mtp config for Hy3 vllm calibration.
1 parent 8bd4dcb commit 2dfca2e

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tools/run_vllm_calibrate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,7 @@ def main():
314314
speculative_config = None
315315
if args.enable_mtp:
316316
speculative_config = {
317-
"method": "hunyuan_mtp",
317+
"method": "mtp",
318318
"num_speculative_tokens": args.num_speculative_tokens,
319319
}
320320
print(f" MTP Enabled: True (num_speculative_tokens={args.num_speculative_tokens})")

0 commit comments

Comments
 (0)