kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-13 22:18:36 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

init

Browse Source 
Signed-off-by: Yan Chunwei <328693+Superjomn@users.noreply.github.com>
This commit is contained in:
Yan Chunwei 2025-10-13 18:17:35 +08:00
parent 206a9930df
commit dda30ec376
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/integration/defs/llmapi/test_llm_e2e.py
View File

@ -90,13 +90,12 @@ def test_llmapi_load_engine_from_build_command_with_lora(
])
@pytest.mark.skip(reason="https://nvbugs/5574355")
@pytest.mark.parametrize("model_name,model_path", [
("llama", "llama-models-v2/TinyLlama-1.1B-Chat-v1.0"),
])
def test_llmapi_build_command_parameters_align(llm_root, llm_venv, engine_dir,
model_name, model_path):
llama_example_root = os.path.join(llm_root, "examples", model_name)
model_name, model_path,
llama_example_root):
dtype = 'float16'
cmodel_dir = os.path.join(engine_dir, f"{model_name}-engine")