"...text-generation-inference.git" did not exist on "2b19d671b4d1020e31276477f278ca87cfa37a3c"
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): ...@@ -668,7 +668,7 @@ def get_dataset(args, tokenizer):
num_prompts=args.num_prompts, num_prompts=args.num_prompts,
range_ratio=args.random_range_ratio, range_ratio=args.random_range_ratio,
tokenizer=tokenizer, tokenizer=tokenizer,
dataset_path=args.dataset_name, dataset_path=args.dataset_path,
random_sample=args.dataset_name == "random", random_sample=args.dataset_name == "random",
return_text=not tokenize_prompt, 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