Skip to content

Commit e46091e

Browse files
committed
Removing skip on GPT2 after local re-tests.
1 parent 977b52b commit e46091e

1 file changed

Lines changed: 0 additions & 4 deletions

File tree

tests/test_e2e.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -44,10 +44,6 @@ class TestE2E:
4444
@pytest.mark.skipif(
4545
os.environ.get("IMAGE_KIND") != "TRTLLM", reason="Only run for TRT-LLM image"
4646
)
47-
@pytest.mark.skipif(
48-
os.environ.get("TRTLLM_MODEL") == "gpt2",
49-
reason="GPT2LMHeadModel architecture is not supported in TensorRT-LLM",
50-
)
5147
@pytest.mark.parametrize(
5248
"protocol",
5349
[

0 commit comments

Comments
 (0)