fix: add seed bench_serving to cache key, remove redundant function definition. (#12680)
This commit is contained in:
@@ -1507,7 +1507,7 @@ def get_gen_prefix_cache_path(args, tokenizer):
|
||||
|
||||
# Create a unique cache filename based on the generation parameters
|
||||
cache_key = (
|
||||
f"gen_shared_prefix_{args.gsp_num_groups}_{args.gsp_prompts_per_group}_"
|
||||
f"gen_shared_prefix_{args.seed}_{args.gsp_num_groups}_{args.gsp_prompts_per_group}_"
|
||||
f"{args.gsp_system_prompt_len}_{args.gsp_question_len}_{args.gsp_output_len}_"
|
||||
f"{tokenizer.__class__.__name__}.pkl"
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user