Skip to content

GitLab

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
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/compile/test_dynamic_shapes_compilation.py
View file @ 612d5ffd
...@@ -77,6 +77,7 @@ def test_dynamic_shapes_compilation( ...@@ -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)
......
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