chatml: minor fix

Lightning-AI · Dec 22, 2024 · b72cc20 · b72cc20
1 parent 1667f29
commit b72cc20
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/litgpt/prompts.py b/litgpt/prompts.py
@@ -317,11 +317,11 @@ def apply(self, prompt: str, **kwargs: str) -> str:
     "gemma": Gemma,
     "llama3": Llama3,
     "olmo": OLMo,
-    "qwen2.5": ChatML("qwen2.5"),
-    "qwen2.5-math": ChatML("qwen2.5-math"),
-    "qwq": ChatML("qwq"),
-    "smollm2": ChatML("smollm2"),
-    "salamandra": ChatML("salamandra"),
+    "qwen2.5": lambda: ChatML("qwen2.5"),
+    "qwen2.5-math": lambda: ChatML("qwen2.5-math"),
+    "qwq": lambda: ChatML("qwq"),
+    "smollm2": lambda: ChatML("smollm2"),
+    "salamandra": lambda: ChatML("salamandra"),
 }