mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
d879430b04
* Update TensorRT-LLM --------- Co-authored-by: Shixiaowei02 <39303645+Shixiaowei02@users.noreply.github.com>
12 lines
341 B
Python
12 lines
341 B
Python
import os
|
|
from pathlib import Path
|
|
|
|
|
|
def llm_models_root() -> str:
|
|
root = Path(
|
|
os.environ.get('LLM_MODELS_ROOT',
|
|
'/home/scratch.trt_llm_data/llm-models/'))
|
|
assert root.exists(), \
|
|
"You shall set LLM_MODELS_ROOT env or be able to access /home/scratch.trt_llm_data to run this test"
|
|
return root
|