diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 414cfe1a..41923092 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -75,11 +75,6 @@ jobs: run: | python -m pip install "transformers==${{ matrix.transformers }}" - - name: Downgrade transformers - if: ${{ matrix.os == 'macos-13' }} - run: | - python -m pip install "transformers<4.52.0" - - name: Cache files id: hf-hub-cache uses: actions/cache@v4 diff --git a/requirements.txt b/requirements.txt index 8cc18753..8206464f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,5 @@ -transformers>=4.45.0,<=4.52.3,!=4.46.*,!=4.47.*,!=4.48.0,!=4.52.0 +transformers>=4.45.0,<=4.52.4,!=4.46.*,!=4.47.*,!=4.48.0,!=4.52.0; sys_platform != 'darwin' +transformers>=4.45.0,<=4.51.3,!=4.46.*,!=4.47.*,!=4.48.0,!=4.52.0; sys_platform == 'darwin' datasets>=2.16.0,<=3.6.0 accelerate>=0.34.0,<=1.7.0 peft>=0.14.0,<=0.15.2 diff --git a/src/llamafactory/extras/misc.py b/src/llamafactory/extras/misc.py index 91794bc7..35b508ff 100644 --- a/src/llamafactory/extras/misc.py +++ b/src/llamafactory/extras/misc.py @@ -95,7 +95,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.45.0,<=4.52.3,!=4.46.0,!=4.46.1,!=4.46.2,!=4.46.3,!=4.47.0,!=4.47.1,!=4.48.0,!=4.52.0" + "transformers>=4.45.0,<=4.52.4,!=4.46.0,!=4.46.1,!=4.46.2,!=4.46.3,!=4.47.0,!=4.47.1,!=4.48.0,!=4.52.0" ) check_version("datasets>=2.16.0,<=3.6.0") check_version("accelerate>=0.34.0,<=1.7.0")