diff --git a/litgpt/prompts.py b/litgpt/prompts.py index a0e515c3f8..34f20ad541 100644 --- 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, }