Update name

hegelai · Jul 25, 2023 · cef64ae · cef64ae
1 parent 3376766
commit cef64ae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/prompttools/selector/prompt_selector.py b/prompttools/selector/prompt_selector.py
@@ -4,7 +4,7 @@
 # This source code's license can be found in the
 # LICENSE file in the root directory of this source tree.
 
-TEMPLATE = """
+LLAMA_TEMPLATE = """
 <s>[INST] <<SYS>>
 {instruction}
 <</SYS>
@@ -30,4 +30,4 @@ def for_openai_chat(self):
  ]
 
  def for_llama(self):
- return TEMPLATE.format(instruction=self.instruction, user_input=self.user_input)
+ return LLAMA_TEMPLATE.format(instruction=self.instruction, user_input=self.user_input)