diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml index 58f9944b08aad..a7d601505d174 100644 --- a/.github/workflows/publish.yml +++ b/.github/workflows/publish.yml @@ -49,7 +49,7 @@ jobs: matrix: os: ['ubuntu-20.04'] python-version: ['3.8', '3.9', '3.10', '3.11'] - pytorch-version: ['2.1.0'] + pytorch-version: ['2.1.1'] cuda-version: ['11.8', '12.1'] steps: diff --git a/pyproject.toml b/pyproject.toml index 045d925cb1131..d18a877420933 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ requires = [ "ninja", "packaging", "setuptools >= 49.4.0", - "torch >= 2.1.0", + "torch >= 2.1.1", "wheel", ] build-backend = "setuptools.build_meta" diff --git a/requirements.txt b/requirements.txt index 9a27eae80973c..5ec101a080e20 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,9 +6,9 @@ pyarrow # Required for Ray data. sentencepiece # Required for LLaMA tokenizer. numpy einops # Required for phi-1_5 -torch >= 2.1.0 +torch >= 2.1.1 transformers >= 4.34.0 # Required for Mistral. -xformers >= 0.0.22.post7 # Required for CUDA 12.1. +xformers >= 0.0.23 # Required for CUDA 12.1. fastapi uvicorn[standard] pydantic == 1.10.13 # Required for OpenAI server.