There was an error while loading. Please reload this page.
1 parent 977b52b commit e46091eCopy full SHA for e46091e
1 file changed
tests/test_e2e.py
@@ -44,10 +44,6 @@ class TestE2E:
44
@pytest.mark.skipif(
45
os.environ.get("IMAGE_KIND") != "TRTLLM", reason="Only run for TRT-LLM image"
46
)
47
- @pytest.mark.skipif(
48
- os.environ.get("TRTLLM_MODEL") == "gpt2",
49
- reason="GPT2LMHeadModel architecture is not supported in TensorRT-LLM",
50
- )
51
@pytest.mark.parametrize(
52
"protocol",
53
[
0 commit comments