mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-08 03:24:07 +00:00
[eval-hotfix] set few_shot_data to None when few shot is disabled (#5422)
This commit is contained in:
parent
049121d19d
commit
a7ae2b5b4c
@ -201,7 +201,7 @@ class AGIEvalDataset(BaseDataset):
|
||||
for file in files:
|
||||
dataset_name = os.path.basename(file)[0 : -len(".jsonl")]
|
||||
|
||||
few_shot_data = []
|
||||
few_shot_data = None
|
||||
if few_shot:
|
||||
# process demo once if it is few-shot-CoT
|
||||
few_shot_data = combine_prompt(prompt_path, dataset_name, load_explanation=False, chat_mode=False)
|
||||
|
Loading…
Reference in New Issue
Block a user