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

Signed-off-by: angelayi <yiangela7@gmail.com>
This commit is contained in:
Angela Yi
2025-12-22 17:56:47 -08:00
committed by GitHub
parent 78e5e62bbf
commit 612d5ffdab

View File

@@ -77,6 +77,7 @@ def test_dynamic_shapes_compilation(
"evaluate_guards": evaluate_guards, "evaluate_guards": evaluate_guards,
}, },
}, },
max_model_len=1024,
) )
output = model.generate(prompt) output = model.generate(prompt)