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

chore: [nvbug_5273941] unwaive test_llm_loading_from_ckpt_for_tp2 (#4725)

Browse Source 
Signed-off-by: Erin Ho <14718778+hchings@users.noreply.github.com>
This commit is contained in:
Erin 2025-05-28 15:54:32 -07:00 committed by GitHub
parent bf691b3d28
commit 820c39041f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/unittest/llmapi/test_llm_multi_gpu.py
View File

@ -77,7 +77,6 @@ def engine_from_checkpoint() -> tempfile.TemporaryDirectory:
@pytest.mark.part0
def test_llm_loading_from_ckpt_for_tp2(
engine_from_checkpoint: tempfile.TemporaryDirectory):
pytest.skip(reason="https://nvbugspro.nvidia.com/bug/5273941")
tokenizer = TransformersTokenizer.from_pretrained(llama_model_path)
llm_test_harness(engine_from_checkpoint.name,
prompts, ["D E F G H I J K"],