Update ChatTemplate enum to include alpaca and gemma (#1396)
Browse files
src/axolotl/utils/config/models/input/v0_4_1/__init__.py
CHANGED
|
@@ -129,8 +129,10 @@ class RLType(str, Enum):
|
|
| 129 |
class ChatTemplate(str, Enum):
|
| 130 |
"""Chat templates configuration subset"""
|
| 131 |
|
|
|
|
| 132 |
chatml = "chatml" # pylint: disable=invalid-name
|
| 133 |
inst = "inst" # pylint: disable=invalid-name
|
|
|
|
| 134 |
|
| 135 |
|
| 136 |
class LoftQConfig(BaseModel):
|
|
@@ -518,7 +520,7 @@ class AxolotlInputConfig(
|
|
| 518 |
] = None
|
| 519 |
gpu_memory_limit: Optional[Union[int, str]] = None
|
| 520 |
|
| 521 |
-
chat_template: Optional[
|
| 522 |
default_system_message: Optional[str] = None
|
| 523 |
|
| 524 |
# INTERNALS - document for now, generally not set externally
|
|
|
|
| 129 |
class ChatTemplate(str, Enum):
|
| 130 |
"""Chat templates configuration subset"""
|
| 131 |
|
| 132 |
+
alpaca = "alpaca" # pylint: disable=invalid-name
|
| 133 |
chatml = "chatml" # pylint: disable=invalid-name
|
| 134 |
inst = "inst" # pylint: disable=invalid-name
|
| 135 |
+
gemma = "gemma" # pylint: disable=invalid-name
|
| 136 |
|
| 137 |
|
| 138 |
class LoftQConfig(BaseModel):
|
|
|
|
| 520 |
] = None
|
| 521 |
gpu_memory_limit: Optional[Union[int, str]] = None
|
| 522 |
|
| 523 |
+
chat_template: Optional[ChatTemplate] = None
|
| 524 |
default_system_message: Optional[str] = None
|
| 525 |
|
| 526 |
# INTERNALS - document for now, generally not set externally
|