Unverified Commit bd95944c authored by Muqi Li's avatar Muqi Li Committed by GitHub
Browse files

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

parent c8a5d12a
......@@ -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,
)
......
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