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

test: Waive test_llm cases (#4136)

Browse Source 
* waive

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>

* Apply suggestions from code review

Signed-off-by: Yanchao Lu <yanchaol@nvidia.com>

---------

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>
Signed-off-by: Yanchao Lu <yanchaol@nvidia.com>
Co-authored-by: Yanchao Lu <yanchaol@nvidia.com>
This commit is contained in:
Enwei Zhu 2025-05-08 11:53:16 +08:00 committed by GitHub
parent 389614ca99
commit 19eeef1ddd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/unittest/llmapi/test_llm.py
View File

@ -228,6 +228,7 @@ def test_llm_loading_from_hf():
kv_cache_config=global_kvcache_config)
@pytest.mark.skip(reason="https://nvbugs/5266240")
@force_ampere
@pytest.mark.part0
def test_llm_loading_from_ckpt():
@ -246,7 +247,11 @@ def test_llm_loading_from_ckpt():
sampling_params=SamplingParams(max_tokens=8))
@pytest.mark.parametrize('model_format', ['hf', 'ckpt'])
@pytest.mark.parametrize('model_format', [
'hf',
pytest.param('ckpt',
marks=pytest.mark.skip(reason="https://nvbugs/5266240"))
])
@pytest.mark.part0
def test_llm_with_dummy_weights(model_format):
# dummy_dir contains config.json and tokenizer files only