Fixing Validation - Mistral Templates (#1962)

This commit is contained in:
pandora
2024-10-13 03:40:39 +02:00
committed by GitHub
parent d20b48a61e
commit 31591bd94c

View File

@@ -187,7 +187,9 @@ class ChatTemplate(str, Enum):
alpaca = "alpaca" # pylint: disable=invalid-name
chatml = "chatml" # pylint: disable=invalid-name
inst = "inst" # pylint: disable=invalid-name
mistral_v1 = "mistral_v1" # pylint: disable=invalid-name
mistral_v2v3 = "mistral_v2v3" # pylint: disable=invalid-name
mistral_v3_tekken = "mistral_v3_tekken" # pylint: disable=invalid-name
gemma = "gemma" # pylint: disable=invalid-name
cohere = "cohere" # pylint: disable=invalid-name
llama3 = "llama3" # pylint: disable=invalid-name