From d58ec6a0bcf07e72ce54197d72ecf36c766adbdd Mon Sep 17 00:00:00 2001 From: Karunanidhi Mishra Date: Tue, 7 Jul 2026 04:30:26 -0500 Subject: [PATCH] [deps] exclude broken transformers release (#10634) Co-authored-by: Karunanidhi Mishra <11963379+kmishra1204@users.noreply.github.com> --- pyproject.toml | 2 +- src/llamafactory/extras/misc.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index e7812cc19..4105361c9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -40,7 +40,7 @@ dependencies = [ "torch>=2.4.0", "torchvision>=0.19.0", "torchaudio>=2.4.0", - "transformers>=4.55.0,<=5.6.0,!=4.57.0", + "transformers>=4.55.0,<=5.7.0,!=4.57.0,!=5.6.0", "datasets>=2.16.0,<=4.0.0", "accelerate>=1.3.0,<=1.11.0", "peft>=0.18.0,<=0.18.1", diff --git a/src/llamafactory/extras/misc.py b/src/llamafactory/extras/misc.py index e128ca418..77137afaf 100644 --- a/src/llamafactory/extras/misc.py +++ b/src/llamafactory/extras/misc.py @@ -94,7 +94,9 @@ 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.55.0,<=5.6.0") + check_version("transformers>=4.55.0,<=5.7.0") + check_version("transformers!=4.57.0") + check_version("transformers!=5.6.0") check_version("datasets>=2.16.0,<=4.0.0") check_version("accelerate>=1.3.0,<=1.15.0") check_version("peft>=0.18.0,<=0.20.0")