423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-24 15:50:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

[misc] fix cache & pin transformers to 4.57.1 (#9638)

Browse Source
This commit is contained in:
Yaowei Zheng
2025-12-22 00:20:55 +08:00
committed by GitHub
parent 4923f52a28
commit 6ef9854713
8 changed files with 35 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1166,7 +1166,7 @@ register_template(
register_template(
name="gpt",
name="gpt_oss",
format_user=StringFormatter(slots=["<|start|>user<|message|>{{content}}<|end|><|start|>assistant"]),
format_assistant=StringFormatter(slots=["{{content}}<|end|>"]),
format_system=StringFormatter(slots=["<|start|>system<|message|>{{content}}<|end|>"]),

14
src/llamafactory/extras/constants.py
View File

@@ -1067,7 +1067,7 @@ register_model_group(
DownloadSource.MODELSCOPE: "openai/gpt-oss-120b",
},
},
template="gpt",
template="gpt_oss",
)
@@ -1995,6 +1995,18 @@ register_model_group(
register_model_group(
models={
"Ministral-3-3B-Base-2512": {
DownloadSource.DEFAULT: "mistralai/Ministral-3-3B-Base-2512",
DownloadSource.MODELSCOPE: "mistralai/Ministral-3-3B-Base-2512",
},
"Ministral-3-8B-Base-2512": {
DownloadSource.DEFAULT: "mistralai/Ministral-3-8B-Base-2512",
DownloadSource.MODELSCOPE: "mistralai/Ministral-3-8B-Base-2512",
},
"Ministral-3-14B-Base-2512": {
DownloadSource.DEFAULT: "mistralai/Ministral-3-14B-Base-2512",
DownloadSource.MODELSCOPE: "mistralai/Ministral-3-14B-Base-2512",
},
"Ministral-3-3B-Instruct-2512": {
DownloadSource.DEFAULT: "mistralai/Ministral-3-3B-Instruct-2512",
DownloadSource.MODELSCOPE: "mistralai/Ministral-3-3B-Instruct-2512",

2
src/llamafactory/extras/misc.py
View File

@@ -94,7 +94,7 @@ def check_version(requirement: str, mandatory: bool = False) -> None:
def check_dependencies() -> None:
r"""Check the version of the required packages."""
check_version("transformers>=4.49.0,<=4.57.3")
check_version("transformers>=4.49.0,<=4.57.1")
check_version("datasets>=2.16.0,<=4.0.0")
check_version("accelerate>=1.3.0,<=1.11.0")
check_version("peft>=0.14.0,<=0.17.1")