support rust sglang server (#29799)
This commit is contained in:
@@ -16,6 +16,7 @@ from sglang.test.simple_eval_common import (
|
||||
ChatCompletionSampler,
|
||||
CompletionSampler,
|
||||
Eval,
|
||||
GenerateSampler,
|
||||
make_report,
|
||||
set_ulimit,
|
||||
)
|
||||
@@ -81,6 +82,13 @@ def run_eval_once(args, base_url: str, eval_obj: Eval) -> dict:
|
||||
**common_kwargs,
|
||||
stop=stop,
|
||||
)
|
||||
elif api_mode == "generate":
|
||||
# SGLang-native `/generate` (raw text + sampling_params), same stop defaults.
|
||||
stop = getattr(args, "stop", ["Question", "Assistant:", "<|separator|>"])
|
||||
sampler = GenerateSampler(
|
||||
**common_kwargs,
|
||||
stop=stop,
|
||||
)
|
||||
else:
|
||||
sampler = ChatCompletionSampler(
|
||||
**common_kwargs,
|
||||
@@ -454,8 +462,8 @@ if __name__ == "__main__":
|
||||
"--api",
|
||||
type=str,
|
||||
default="chat",
|
||||
choices=["chat", "completion"],
|
||||
help="API mode: 'chat' for /v1/chat/completions, 'completion' for /v1/completions",
|
||||
choices=["chat", "completion", "generate"],
|
||||
help="API mode: 'chat' for /v1/chat/completions, 'completion' for /v1/completions, 'generate' for SGLang-native /generate",
|
||||
)
|
||||
parser.add_argument("--num-examples", type=int)
|
||||
parser.add_argument("--num-threads", type=int, default=512)
|
||||
|
||||
Reference in New Issue
Block a user