ros/LlamaFactory
1
0
Fork 0
mirror of https://github.com/hiyouga/LlamaFactory.git synced 2026-02-02 08:33:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

[misc] fix constants (#9008)

Browse Source
This commit is contained in:
Yaowei Zheng
2025-08-23 23:04:30 +08:00
committed by GitHub
parent 0ab0be9df2
commit ec41ef08aa
6 changed files with 40 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/data/formatter.py
View File

@@ -98,7 +98,7 @@ class FunctionFormatter(StringFormatter):
def apply(self, **kwargs) -> SLOTS:
content: str = kwargs.pop("content")
thought_words, thought = kwargs.pop("thought_words", None), None
if thought_words and len(thought_words)== 2:
if thought_words and len(thought_words) == 2:
regex = re.compile(rf"{re.escape(thought_words[0])}(.*?){re.escape(thought_words[1])}", re.DOTALL)
thought = re.search(regex, content)