From cec98876090bb9610437bf224fac736752ca968e Mon Sep 17 00:00:00 2001 From: bursteratom Date: Fri, 6 Dec 2024 12:57:20 -0500 Subject: [PATCH] add llava chat template to config --- src/axolotl/utils/config/models/input/v0_4_1/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py index 51e1eefc1..ed1eb98b0 100644 --- a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py +++ b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py @@ -51,6 +51,7 @@ class ChatTemplate(str, Enum): cohere = "cohere" # pylint: disable=invalid-name llama3 = "llama3" # pylint: disable=invalid-name llama3_2_vision = "llama3_2_vision" # pylint: disable=invalid-name + llava = "llava" # pylint: disable=invalid-name phi_3 = "phi_3" # pylint: disable=invalid-name phi_35 = "phi_35" # pylint: disable=invalid-name deepseek_v2 = "deepseek_v2" # pylint: disable=invalid-name