tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Fixing Validation - Mistral Templates (#1962)

Browse Source
This commit is contained in:
pandora
2024-10-13 03:40:39 +02:00
committed by GitHub
parent d20b48a61e
commit 31591bd94c
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/axolotl/utils/config/models/input/v0_4_1/__init__.py
View File

@@ -187,7 +187,9 @@ class ChatTemplate(str, Enum):
alpaca = "alpaca" # pylint: disable=invalid-name alpaca = "alpaca" # pylint: disable=invalid-name
chatml = "chatml" # 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 gemma = "gemma" # pylint: disable=invalid-name
cohere = "cohere" # pylint: disable=invalid-name cohere = "cohere" # pylint: disable=invalid-name
llama3 = "llama3" # pylint: disable=invalid-name llama3 = "llama3" # pylint: disable=invalid-name