Add zephyr-alpha templates

Cyrilvallez · Dec 5, 2023 · 91082df · 91082df
1 parent 0f6489d
commit 91082df
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/textwiz/conversation_template.py b/textwiz/conversation_template.py
@@ -494,6 +494,7 @@ def get_prompt(self) -> str:
     'mistral-7B-instruct': MistralConversation,
 
     # Zephyr
+    'zephyr-7B-alpha': ZephyrConversation,
     'zephyr-7B-beta': ZephyrConversation,
 }
 

diff --git a/textwiz/prompt_template.py b/textwiz/prompt_template.py
@@ -386,6 +386,7 @@ def format_chat(self, prompt: str, model_context: str = '', system_prompt: str =
     'mistral-7B-instruct': MistralPromptTemplate,
 
     # Zephyr
+    'zephyr-7B-alpha': ZephyrPromptTemplate,
     'zephyr-7B-beta': ZephyrPromptTemplate,
 }