EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Tiny fix CI (#6611)

Browse Source
This commit is contained in:
fzyzcjy
2025-05-26 14:36:34 +08:00
committed by GitHub
parent f9bab3d591
commit 501efc3d36
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
python/sglang/bench_serving.py
View File

@@ -493,8 +493,9 @@ def get_tokenizer(
def get_dataset(args, tokenizer): def get_dataset(args, tokenizer):
tokenize_prompt = getattr(args, "tokenize_prompt", False)
if args.dataset_name == "sharegpt": if args.dataset_name == "sharegpt":
assert not args.tokenize_prompt assert not tokenize_prompt
input_requests = sample_sharegpt_requests( input_requests = sample_sharegpt_requests(
dataset_path=args.dataset_path, dataset_path=args.dataset_path,
num_requests=args.num_prompts, num_requests=args.num_prompts,
@@ -513,10 +514,10 @@ def get_dataset(args, tokenizer):
tokenizer=tokenizer, tokenizer=tokenizer,
dataset_path=args.dataset_path, dataset_path=args.dataset_path,
random_sample=args.dataset_name == "random", random_sample=args.dataset_name == "random",
return_text=not args.tokenize_prompt, return_text=not tokenize_prompt,
) )
elif args.dataset_name == "generated-shared-prefix": elif args.dataset_name == "generated-shared-prefix":
assert not args.tokenize_prompt assert not tokenize_prompt
input_requests = sample_generated_shared_prefix_requests( input_requests = sample_generated_shared_prefix_requests(
num_groups=args.gsp_num_groups, num_groups=args.gsp_num_groups,
prompts_per_group=args.gsp_prompts_per_group, prompts_per_group=args.gsp_prompts_per_group,
@@ -527,7 +528,7 @@ def get_dataset(args, tokenizer):
args=args, args=args,
) )
elif args.dataset_name == "mmmu": elif args.dataset_name == "mmmu":
assert not args.tokenize_prompt assert not tokenize_prompt
input_requests = sample_mmmu_requests( input_requests = sample_mmmu_requests(
num_requests=args.num_prompts, num_requests=args.num_prompts,
tokenizer=tokenizer, tokenizer=tokenizer,