Skip to content

GitLab

Commit bdb14434 authored by zihanl's avatar zihanl
Browse files

update prompt

parent a79820a6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tasks/knwl_dialo/prompt.py
View file @ bdb14434
...@@ -104,7 +104,7 @@ def generate_samples_by_prompting_input_from_file(model): ...@@ -104,7 +104,7 @@ def generate_samples_by_prompting_input_from_file(model):
if args.prompt_type == "knowledge": if args.prompt_type == "knowledge":
turns = splits[1].split(" [SEP] ") turns = splits[1].split(" [SEP] ")
context = turns[-1] context = turns[-1]
if " -> " in raw_text: if " -> " in raw_text and " => " not in raw_text:
raw_text += "( " + context + " ) " + topic + " ->" raw_text += "( " + context + " ) " + topic + " ->"
else: else:
raw_text += "( " + context + " ) " + topic + " =>" 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