From b72cc208a6bcd583da6b2513bf98e49e7c9424cd Mon Sep 17 00:00:00 2001
From: Yu Shi Jie <yshijie1999@gmail.com>
Date: Sun, 22 Dec 2024 11:23:54 -0500
Subject: [PATCH] chatml: minor fix

---
 litgpt/prompts.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/litgpt/prompts.py b/litgpt/prompts.py
index 42268dc45f..a438b60ca5 100644
--- 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"),
 }