From 43052327af849661ef85f866a94c3084f0bba9e4 Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Sun, 27 Oct 2024 14:31:51 +0100 Subject: [PATCH] fix --- .github/workflows/test_generation.yml | 2 +- .github/workflows/test_ipex.yml | 2 +- setup.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/test_generation.yml b/.github/workflows/test_generation.yml index 3bc04e3c6..345a134ae 100644 --- a/.github/workflows/test_generation.yml +++ b/.github/workflows/test_generation.yml @@ -1,4 +1,4 @@ -name: Generation Utils - Test +name: Generation Utils - Test (deprecated) on: push: diff --git a/.github/workflows/test_ipex.yml b/.github/workflows/test_ipex.yml index b8ea66387..48efe9daf 100644 --- a/.github/workflows/test_ipex.yml +++ b/.github/workflows/test_ipex.yml @@ -40,7 +40,7 @@ jobs: pip install torch==${{ matrix.torch-version }} torchaudio torchvision --extra-index-url https://download.pytorch.org/whl/cpu pip install intel_extension_for_pytorch==${{ matrix.torch-version }} pip install transformers==${{ matrix.transformers-version }} - pip install .[ipex] + pip install .[ipex,tests] - name: Assert versions run: | diff --git a/setup.py b/setup.py index 95c99af23..415cd8886 100644 --- a/setup.py +++ b/setup.py @@ -28,7 +28,7 @@ INSTALL_REQUIRE = [ "torch>=1.11", - "transformers>=4.36,<4.46", + "transformers>=4.36.0,<4.46.0", "optimum~=1.23", "datasets>=1.4.0", "sentencepiece",