Skip to content

Commit f345ecf

Browse files
committed
Fix llama4 disaggregated serving accuracy tests
Signed-off-by: Iman Tabrizian <[email protected]>
1 parent cdde37b commit f345ecf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/integration/defs/accuracy/test_disaggregated_serving.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@ def launch_disaggregated_llm(disaggregated_server_config: Dict[str, Any],
8383
yaml.dump(gen_server_config, f)
8484

8585
args = LlmArgs.from_kwargs(model=model_name,
86+
backend="pytorch",
8687
tensor_parallel_size=tensor_parallel_size)
8788

8889
trtllm_serve_path = "trtllm-serve"
@@ -219,7 +220,6 @@ class TestLlama4ScoutInstruct(LlmapiAccuracyTestHarness):
219220

220221
@pytest.mark.parametrize("overlap_scheduler", [False, True])
221222
def test_auto_dtype(self, overlap_scheduler):
222-
pytest.skip("https://nvbugs/5297821")
223223
ctx_server_config = {
224224
"pytorch_backend_config": {
225225
"disable_overlap_scheduler": True

0 commit comments

Comments
 (0)