diff --git a/src/autolabel/tasks/base.py b/src/autolabel/tasks/base.py index 0b3f636a..f21af7bc 100644 --- a/src/autolabel/tasks/base.py +++ b/src/autolabel/tasks/base.py @@ -34,12 +34,12 @@ class BaseTask(ABC): ZERO_SHOT_TEMPLATE_REFUEL_LLM = """ [INST] <> {task_guidelines}{output_guidelines} - <> + <> {current_example}[/INST]\n""" FEW_SHOT_TEMPLATE_REFUEL_LLM = """ [INST] <> {task_guidelines}{output_guidelines}\n{seed_examples} - <> + <> {current_example}[/INST]\n""" # Downstream classes should override these