alfraser commited on
Commit
3991f6c
·
1 Parent(s): ee60fb2

Trying a different prompting style

Browse files
Files changed (1) hide show
  1. src/training/prep_finetuning.py +1 -3
src/training/prep_finetuning.py CHANGED
@@ -180,9 +180,7 @@ def training_string_from_q_and_a(q: str, a: str, sys_prompt: str = None) -> str:
180
  Build the single llama formatted training string from a question
181
  answer pair
182
  """
183
- if sys_prompt is None:
184
- sys_prompt = "You are a helpful domestic appliance advisor for the ElectroHome company. Please answer customer questions and do not mention other brands. If you cannot answer please say so."
185
- return f"<s>[INST] <<SYS>>\n{sys_prompt}\n<</SYS>>\n\n{q} [/INST]{a}"
186
 
187
 
188
  def fine_tuning_out_dir(out_model: str) -> str:
 
180
  Build the single llama formatted training string from a question
181
  answer pair
182
  """
183
+ return f"<s>[INST] {q} [/INST] {a}"
 
 
184
 
185
 
186
  def fine_tuning_out_dir(out_model: str) -> str: