From abbf15491472367db247302ae99b7e0da2553c9d Mon Sep 17 00:00:00 2001 From: Morgan Funtowicz Date: Tue, 25 Jun 2024 16:23:25 +0200 Subject: [PATCH] feat(trtllm): Force TRTLLM to 0.11.0.dev2024062500 --- pyproject.toml | 2 +- setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index fd65052d..4721175c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -33,7 +33,7 @@ dependencies = [ "onnx >= 1.12.0", "optimum >= 1.13.0", "setuptools", - "tensorrt-llm > 0.10.0", + "tensorrt-llm == 0.11.0.dev2024062500", "torch>=2.2.0a,<=2.3.0a", "transformers >= 4.38.2", "pynvml" diff --git a/setup.py b/setup.py index 931e722d..191effc4 100644 --- a/setup.py +++ b/setup.py @@ -36,7 +36,7 @@ "onnx >= 1.12.0", "optimum >= 1.13.0", "setuptools", - "tensorrt-llm > 0.10.0", + "tensorrt-llm == 0.11.0.dev2024062500", "torch>=2.2.0a,<=2.3.0a", "transformers >= 4.38.2", "pynvml"