Unverified Commit 612d5ffd authored by Angela Yi's avatar Angela Yi Committed by GitHub
Browse files

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


Signed-off-by: default avatarangelayi <yiangela7@gmail.com>
parent 78e5e62b
......@@ -77,6 +77,7 @@ def test_dynamic_shapes_compilation(
"evaluate_guards": evaluate_guards,
},
},
max_model_len=1024,
)
output = model.generate(prompt)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment