Commit bdb14434 authored by zihanl's avatar zihanl
Browse files

update prompt

parent a79820a6
......@@ -104,7 +104,7 @@ def generate_samples_by_prompting_input_from_file(model):
if args.prompt_type == "knowledge":
turns = splits[1].split(" [SEP] ")
context = turns[-1]
if " -> " in raw_text:
if " -> " in raw_text and " => " not in raw_text:
raw_text += "( " + context + " ) " + topic + " ->"
else:
raw_text += "( " + context + " ) " + topic + " =>"
......
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