"tests/python/common/test_subgraph.py" did not exist on "223a3da5aede2ec555289af760888e4b50fd2de4"
Unverified Commit 16ff3d4b authored by wenhuipeng's avatar wenhuipeng Committed by GitHub
Browse files

Support opt model (#10165)

parent 83d55ac5
This diff is collapsed.
...@@ -77,6 +77,7 @@ ALL_MODELS = [ ...@@ -77,6 +77,7 @@ ALL_MODELS = [
trust_remote_code=True, trust_remote_code=True,
skip_long_prompt=True, skip_long_prompt=True,
), ),
ModelCase("facebook/opt-125m", skip_long_prompt=True),
ModelCase( ModelCase(
"nvidia/Llama-3_3-Nemotron-Super-49B-v1_5", "nvidia/Llama-3_3-Nemotron-Super-49B-v1_5",
tp_size=2, tp_size=2,
......
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