kyy/llm_trainer
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

support llama pro #2338 , add rslora

Browse Source
This commit is contained in:
hiyouga committed 2024-02-15 02:27:36 +08:00
1 parent 8a1b389086
commit 7924ffc55d
24 files changed
+437 -202
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/llmtuner/data/utils.py
+1 -1
View File
@@ -20,8 +20,8 @@ class Role(str, Enum):
USER = "user"
ASSISTANT = "assistant"
SYSTEM = "system"
OBSERVATION = "observation"
FUNCTION = "function"
OBSERVATION = "observation"
def checksum(data_files: List[str], file_sha1: Optional[str] = None) -> None: