423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-11-04 09:52:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[assets] update wechat (#9129)

Browse Source
This commit is contained in:
Yaowei Zheng 2025-09-14 03:05:08 +08:00 committed by GitHub
parent 52488ac974
commit 260b5625c3
3 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/llamafactory/data/converter.py
View File

@ -257,8 +257,8 @@ class OpenAIDatasetConverter(DatasetConverter):
content = message[self.dataset_attr.content_tag]
if role in [self.dataset_attr.assistant_tag, self.dataset_attr.function_tag]:
if "tool_calls" in message and len(message['tool_calls']) > 0:
tool_calls_list = [tool['function'] for tool in message['tool_calls']]
if "tool_calls" in message and len(message["tool_calls"]) > 0:
tool_calls_list = [tool["function"] for tool in message["tool_calls"]]
content = json.dumps(tool_calls_list, ensure_ascii=False)
role = self.dataset_attr.function_tag
@ -340,7 +340,7 @@ class OpenAIDatasetConverter(DatasetConverter):
if isinstance(tools, dict) or isinstance(tools, list):
tools = json.dumps(tools, ensure_ascii=False)
short_system_prompt = 'detailed thinking off'
short_system_prompt = "detailed thinking off"
if not system:
if not tools:
system = short_system_prompt
@ -348,13 +348,12 @@ class OpenAIDatasetConverter(DatasetConverter):
pass
else:
if not tools:
if 'detailed thinking on' in system or 'detailed thinking off' in system:
if "detailed thinking on" in system or "detailed thinking off" in system:
pass
else:
system += '\n' + short_system_prompt
system += "\n" + short_system_prompt
else:
system += '\n'
system += "\n"
output = {
"_prompt": prompt,

4
src/llamafactory/data/template.py
View File

@ -685,7 +685,9 @@ register_template(
format_system=StringFormatter(slots=["<role>SYSTEM</role>{{content}}<|role_end|>"]),
format_assistant=StringFormatter(slots=["{{content}}<|role_end|>"]),
format_observation=StringFormatter(
slots=["<role>OBSERVATION</role>\n<tool_response>\n{{content}}\n</tool_response><|role_end|><role>ASSISTANT</role>"]
slots=[
"<role>OBSERVATION</role>\n<tool_response>\n{{content}}\n</tool_response><|role_end|><role>ASSISTANT</role>"
]
),
format_function=FunctionFormatter(slots=["{{content}}<|role_end|>"], tool_format="ling"),
format_tools=ToolFormatter(tool_format="ling"),

2
src/llamafactory/model/patcher.py
View File

@ -219,7 +219,7 @@ def patch_valuehead_model(model: "AutoModelForCausalLMWithValueHead") -> None:
if base_model and hasattr(base_model, "get_rope_index"):
return base_model.get_rope_index
elif (base_model and hasattr(base_model, "model") and hasattr(base_model.model, "get_rope_index")):
elif base_model and hasattr(base_model, "model") and hasattr(base_model.model, "get_rope_index"):
return base_model.model.get_rope_index
else:
return None