Skip to content

Commit 55ccfa5

Browse files
whoisjmc-nv
authored andcommitted
fix: Update L0_openai to adapt TRT-LLM v1.0 (#8430)
1 parent e2cbad2 commit 55ccfa5

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

qa/L0_openai/generate_engine.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,8 @@
2525
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
2626
from argparse import ArgumentParser
2727

28-
from tensorrt_llm import LLM, BuildConfig
28+
from tensorrt_llm import BuildConfig
29+
from tensorrt_llm._tensorrt_engine import LLM
2930
from tensorrt_llm.plugin import PluginConfig
3031

3132

0 commit comments

Comments
 (0)