[deps] exclude broken transformers release (#10634)

Co-authored-by: Karunanidhi Mishra <11963379+kmishra1204@users.noreply.github.com>
This commit is contained in:
Karunanidhi Mishra
2026-07-07 04:30:26 -05:00
committed by GitHub
parent 5987a8dd68
commit d58ec6a0bc
2 changed files with 4 additions and 2 deletions

View File

@@ -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",

View File

@@ -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")