Add prompt style mapping for llama3 (#1406)

Lightning-AI · May 20, 2024 · cbbe9cd · cbbe9cd
1 parent 17e5c77
commit cbbe9cd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/litgpt/prompts.py b/litgpt/prompts.py
@@ -318,6 +318,7 @@ def apply(self, prompt: str, **kwargs: str) -> str:
     "tinyllama": TinyLlama,
     "gemma": Gemma,
     "h2oai": H2Oai,
+    "llama3": Llama3,
 }