prompt fix

Lightning-AI · Apr 13, 2024 · ea5f9bf · ea5f9bf
1 parent 3539d27
commit ea5f9bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()