mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
fix yapf
This commit is contained in:
parent
2450a22f18
commit
ea517e3377
@ -111,7 +111,12 @@ class GenInferencerOutputHandler:
|
||||
"""Dump the result to a json file."""
|
||||
dump_results_dict(self.results_dict, Path(save_dir) / filename)
|
||||
|
||||
def save_results(self, origin_prompt, prediction, idx, api_prompts=None, gold=None):
|
||||
def save_results(self,
|
||||
origin_prompt,
|
||||
prediction,
|
||||
idx,
|
||||
api_prompts=None,
|
||||
gold=None):
|
||||
results = {}
|
||||
if api_prompts:
|
||||
results['api_prompts'] = api_prompts
|
||||
|
@ -156,15 +156,15 @@ class GenInferencer(BaseInferencer):
|
||||
api_prompts_list = parsed_entries
|
||||
prompts, generated = generated
|
||||
else:
|
||||
api_prompts_list = [None]*len(generated)
|
||||
api_prompts_list = [None] * len(generated)
|
||||
prompts = parsed_entries
|
||||
|
||||
num_return_sequences = getattr(self.model, 'generation_kwargs',
|
||||
{}).get('num_return_sequences', 1)
|
||||
# 5-3. Save current output
|
||||
for api_prompts, prompt, prediction, gold in zip(
|
||||
api_prompts_list, prompts, batched(generated, num_return_sequences),
|
||||
golds):
|
||||
api_prompts_list, prompts,
|
||||
batched(generated, num_return_sequences), golds):
|
||||
if num_return_sequences == 1:
|
||||
prediction = prediction[0]
|
||||
output_handler.save_results(prompt,
|
||||
|
Loading…
Reference in New Issue
Block a user