From c83b759093ef6d235a2591a18e4889812109cf93 Mon Sep 17 00:00:00 2001 From: Ivy Zhang <25222398+crazydemo@users.noreply.github.com> Date: Tue, 19 Aug 2025 16:30:08 +0800 Subject: [PATCH] fix import error Signed-off-by: Ivy Zhang <25222398+crazydemo@users.noreply.github.com> --- tests/integration/defs/examples/test_llama.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/integration/defs/examples/test_llama.py b/tests/integration/defs/examples/test_llama.py index d63737dbe0c..60505a4d56c 100644 --- a/tests/integration/defs/examples/test_llama.py +++ b/tests/integration/defs/examples/test_llama.py @@ -4135,7 +4135,8 @@ def test_llm_api_lookahead_decoding_1gpu(model_name, model_path): """ from defs.conftest import llm_models_root - from tensorrt_llm.llmapi import (LLM, BuildConfig, KvCacheConfig, + from tensorrt_llm._tensorrt_engine import LLM + from tensorrt_llm.llmapi import (BuildConfig, KvCacheConfig, LookaheadDecodingConfig, SamplingParams) build_config = BuildConfig(max_batch_size=128, max_input_len=2048,