Salamandra: fix chat template only applies to instruct

Lightning-AI · Dec 6, 2024 · 2f35d56 · 2f35d56
1 parent 7cfc086
commit 2f35d56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/litgpt/prompts.py b/litgpt/prompts.py
@@ -364,7 +364,7 @@ def model_name_to_prompt_style(model_name: str) -> PromptStyle:
         return Qwen2_5()
     if re.search(r"QwQ-.*", model_name):
         return QwQ()
-    if re.search(r"salamandra-.*", model_name):
+    if re.search(r"salamandra-.*-instruct", model_name):
         return Salamandra()
     return Default()