[ci] Fix Pytorch compilation test oom in 2.10 (#31194)
Signed-off-by: angelayi <yiangela7@gmail.com>
This commit is contained in:
@@ -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)
|
||||||
|
|||||||
Reference in New Issue
Block a user