kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-02-05 18:51:38 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[https://nvbugs/5836592][fix] Fix qwen3 eagle test (#11030)

Browse Source 
Signed-off-by: Mike Iovine <miovine@nvidia.com>
This commit is contained in:
Mike Iovine 2026-01-29 17:49:08 -05:00 committed by GitHub
parent a4880ffdbb
commit 0ad87895f5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/integration/defs/accuracy/test_llm_api_pytorch.py
View File

@ -3720,8 +3720,8 @@ class TestQwen3_4B(LlmapiAccuracyTestHarness):
target_model_dir = f"{llm_models_root()}/Qwen3/Qwen3-4B"
draft_len = 3
spec_config = EagleDecodingConfig(max_draft_len=draft_len,
speculative_model_dir=eagle_model_dir)
spec_config = Eagle3DecodingConfig(
max_draft_len=draft_len, speculative_model_dir=eagle_model_dir)
with LLM(model=target_model_dir,
**pytorch_config,