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

[deps] upgrade transformers to 4.56.1 (#9128)

Browse Source
This commit is contained in:
Yaowei Zheng 2025-09-14 02:26:39 +08:00 committed by GitHub
parent b95c11d8ea
commit cf48406d07
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 18 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.github/workflows/tests.yml vendored
View File

@ -40,6 +40,9 @@ jobs:
- python: "3.9" - python: "3.9"
os: "ubuntu-latest" os: "ubuntu-latest"
transformers: "4.51.0" transformers: "4.51.0"
- python: "3.9"
os: "ubuntu-latest"
transformers: "4.53.0"
runs-on: ${{ matrix.os }} runs-on: ${{ matrix.os }}

11
requirements.txt
View File

@ -1,12 +1,11 @@
# core deps # core deps
transformers>=4.49.0,<=4.55.0,!=4.52.0 transformers>=4.49.0,<=4.56.1,!=4.52.0
datasets>=2.16.0,<=3.6.0 datasets>=2.16.0,<=4.0.0
accelerate>=1.3.0,<=1.7.0 accelerate>=1.3.0,<=1.10.1
peft>=0.14.0,<=0.15.2 peft>=0.14.0,<=0.17.1
trl>=0.8.6,<=0.9.6 trl>=0.8.6,<=0.9.6
tokenizers>=0.19.0,<=0.21.1
# gui # gui
gradio>=4.38.0,<=5.42.0 gradio>=4.38.0,<=5.45.0
matplotlib>=3.7.0 matplotlib>=3.7.0
tyro<0.9.0 tyro<0.9.0
# ops # ops

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

@ -94,10 +94,10 @@ def check_version(requirement: str, mandatory: bool = False) -> None:
def check_dependencies() -> None: def check_dependencies() -> None:
r"""Check the version of the required packages.""" r"""Check the version of the required packages."""
check_version("transformers>=4.49.0,<=4.55.0") check_version("transformers>=4.49.0,<=4.56.1")
check_version("datasets>=2.16.0,<=3.6.0") check_version("datasets>=2.16.0,<=4.0.0")
check_version("accelerate>=1.3.0,<=1.7.0") check_version("accelerate>=1.3.0,<=1.10.1")
check_version("peft>=0.14.0,<=0.15.2") check_version("peft>=0.14.0,<=0.17.1")
check_version("trl>=0.8.6,<=0.9.6") check_version("trl>=0.8.6,<=0.9.6")

4
src/llamafactory/extras/packages.py
View File

@ -58,6 +58,10 @@ def is_apollo_available():
return _is_package_available("apollo_torch") return _is_package_available("apollo_torch")
def is_jieba_available():
return _is_package_available("jieba")
def is_gradio_available(): def is_gradio_available():
return _is_package_available("gradio") return _is_package_available("gradio")

4
src/llamafactory/train/sft/metric.py
View File

@ -21,11 +21,11 @@ from typing import TYPE_CHECKING, Optional
import numpy as np import numpy as np
import torch import torch
from transformers.utils import is_jieba_available, is_nltk_available from transformers.utils import is_nltk_available
from ...extras.constants import IGNORE_INDEX from ...extras.constants import IGNORE_INDEX
from ...extras.misc import numpify from ...extras.misc import numpify
from ...extras.packages import is_rouge_available from ...extras.packages import is_jieba_available, is_rouge_available
if TYPE_CHECKING: if TYPE_CHECKING: