Add CI for gpt-oss model on hopper (#8851)

This commit is contained in:
fzyzcjy
2025-08-09 15:34:23 +08:00
committed by GitHub
parent de8b8b6e5c
commit 442534aa44
7 changed files with 187 additions and 2 deletions

View File

@@ -65,9 +65,10 @@ def run_eval(args):
sampler = ChatCompletionSampler(
model=args.model,
max_tokens=2048,
max_tokens=getattr(args, "max_tokens", 2048),
base_url=base_url,
temperature=getattr(args, "temperature", 0.0),
reasoning_effort=getattr(args, "reasoning_effort", None),
)
# Run eval
@@ -120,7 +121,9 @@ if __name__ == "__main__":
parser.add_argument("--eval-name", type=str, default="mmlu")
parser.add_argument("--num-examples", type=int)
parser.add_argument("--num-threads", type=int, default=512)
parser.add_argument("--max-tokens", type=int, default=2048)
parser.add_argument("--temperature", type=float, default=0.0)
parser.add_argument("--reasoning-effort", type=str)
args = parser.parse_args()
run_eval(args)