Skip to content

Commit 5f80ebd

Browse files
committed
disable precompiled when TRTLLM_USE_PRECOMPILED=0
Signed-off-by: Jiagan Cheng <[email protected]>
1 parent 9f82262 commit 5f80ebd

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

setup.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,9 @@ def extract_from_precompiled(precompiled_location: str, package_data: List[str],
203203

204204
precompiled: str | None = os.getenv("TRTLLM_USE_PRECOMPILED")
205205
precompiled_location: str | None = os.getenv("TRTLLM_PRECOMPILED_LOCATION")
206-
use_precompiled: bool = precompiled is not None or precompiled_location is not None
206+
use_precompiled: bool = (precompiled is not None
207+
and precompiled != "0") or (precompiled_location
208+
is not None)
207209

208210
if use_precompiled:
209211
from tempfile import TemporaryDirectory

0 commit comments

Comments
 (0)