[Bugfix][Minor][Benchmark] Fix some bugs due to PR #10495 (#10982)

This commit is contained in:
Muqi Li
2025-09-27 13:01:05 +08:00
committed by GitHub
parent c8a5d12abe
commit bd95944cf6

View File

@@ -668,7 +668,7 @@ def get_dataset(args, tokenizer):
num_prompts=args.num_prompts,
range_ratio=args.random_range_ratio,
tokenizer=tokenizer,
dataset_path=args.dataset_name,
dataset_path=args.dataset_path,
random_sample=args.dataset_name == "random",
return_text=not tokenize_prompt,
)