From ea5f9bf3c7abc0b23095350b41c0443057fd5810 Mon Sep 17 00:00:00 2001
From: Dev Khant <devkhant24@gmail.com>
Date: Sat, 13 Apr 2024 11:34:48 +0530
Subject: [PATCH] prompt fix

---
 litgpt/prompts.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/litgpt/prompts.py b/litgpt/prompts.py
index f7fd59d0a0..df38aa3937 100644
--- a/litgpt/prompts.py
+++ b/litgpt/prompts.py
@@ -339,7 +339,7 @@ def model_name_to_prompt_style(model_name: str) -> PromptStyle:
     if re.search(r"(Code)?Gemma.*-it", model_name):
         return Gemma()
     if re.search("Danube2.*-chat", model_name):
-        return H2Oai
+        return H2Oai()
     return Default()