improve aligner

2024-02-10 16:39:19 +08:00
parent 388b705a8d
commit 7d2dc83c5e
11 changed files with 127 additions and 112 deletions
--- a/src/llmtuner/data/utils.py
+++ b/src/llmtuner/data/utils.py
@@ -19,9 +19,9 @@ logger = get_logger(__name__)
 class Role(str, Enum):
    USER = "user"
    ASSISTANT = "assistant"
+    SYSTEM = "system"
    OBSERVATION = "observation"
    FUNCTION = "function"
-    SYSTEM = "system"


 def checksum(data_files: List[str], file_sha1: Optional[str] = None) -> None: