cherry pick #5416 (#5776)

Signed-off-by: junq <[email protected]>

Author: QiJune

tests/unittest/_torch/test_pytorch_model_engine.py

@@ -1,7 +1,6 @@
 import unittest
 from dataclasses import dataclass
 
-import pytest
 import torch
 
 import tensorrt_llm
@@ -43,6 +42,9 @@ def __init__(self, dtype: torch.dtype):
             torch_dtype=dtype))
         self.recorded_position_ids = None
 
+    def infer_max_seq_len(self):
+        return 2048
+
     @property
     def config(self):
         return self.model_config.pretrained_config
@@ -133,7 +135,6 @@ def create_model_engine_and_kvcache(config: PyTorchConfig = None):
     return model_engine, kv_cache_manager
 
 
-@pytest.mark.skip(reason="https://nvbugs/5324248")
 class PyTorchModelEngineTestCase(unittest.TestCase):
 
     def test_pad_generation_requests(self) -> None: