[ci] Fix Pytorch compilation test oom in 2.10 (#31194)

Signed-off-by: angelayi <yiangela7@gmail.com>
2025-12-22 17:56:47 -08:00
parent 78e5e62bbf
commit 612d5ffdab
1 changed files with 1 additions and 0 deletions
--- a/tests/compile/test_dynamic_shapes_compilation.py
+++ b/tests/compile/test_dynamic_shapes_compilation.py
@@ -77,6 +77,7 @@ def test_dynamic_shapes_compilation(
                "evaluate_guards": evaluate_guards,
            },
        },
+        max_model_len=1024,
    )

    output = model.generate(prompt)