Commit c314246d authored by momotori's avatar momotori
Browse files

fix: _generate_configs.py

parent a7707c76
......@@ -47,10 +47,9 @@ if __name__ == "__main__":
prefix_doc_to_text = ""
if args.fewshot:
if args.cot:
prefix_doc_to_text = " ".join(few_shot)
prefix_doc_to_text = "\n\n".join(few_shot) + "\n\n"
else:
for shot in few_shot:
shot = "Q:" + shot
try:
answer = answer_regex.search(shot)[0]
except Exception:
......
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