diff --git a/.github/workflows/nightly_publish.yml b/.github/workflows/nightly_publish.yml index 0478e2fb7d5..c9cb66fc9b9 100644 --- a/.github/workflows/nightly_publish.yml +++ b/.github/workflows/nightly_publish.yml @@ -176,10 +176,12 @@ jobs: - name: Publish to snapshot repository if: ${{ github.event.inputs.mode == '' || github.event.inputs.mode == 'snapshot' }} # PT 1.13.1 is for PT 1.x support, and CUDA 11.x support + # PT 2.1.2 is for Neuron support (latest supported version on neuron) # PT 2.3.1 is for CUDA 12.1 support # PT 2.5.1 is for CUDA 12.4 support run: | ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=1.13.1 -Psnapshot + ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.1.2 -Psnapshot ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.3.1 -Psnapshot ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.5.1 -Psnapshot ./gradlew clean engines:ml:xgboost:publish -Pgpu -Psnapshot @@ -197,7 +199,6 @@ jobs: ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=1.13.1 -P${{ github.event.inputs.mode }} ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.1.2 -P${{ github.event.inputs.mode }} ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.3.1 -P${{ github.event.inputs.mode }} - ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.4.0 -P${{ github.event.inputs.mode }} ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.5.1 -P${{ github.event.inputs.mode }} ./gradlew clean engines:ml:xgboost:publish -Pgpu -P${{ github.event.inputs.mode }} ./gradlew clean publish -P${{ github.event.inputs.mode }} diff --git a/engines/pytorch/pytorch-engine/README.md b/engines/pytorch/pytorch-engine/README.md index 23e42d41fa3..4623fd0329a 100644 --- a/engines/pytorch/pytorch-engine/README.md +++ b/engines/pytorch/pytorch-engine/README.md @@ -51,7 +51,7 @@ The following table illustrates which pytorch version that DJL supports: | PyTorch engine version | PyTorch native library version | |------------------------|-------------------------------------------| -| pytorch-engine:0.32.0 | 1.13.1, 2.3.1, **2.5.1** | +| pytorch-engine:0.32.0 | 1.13.1, 2.1.2, 2.3.1, **2.5.1** | | pytorch-engine:0.31.0 | 1.13.1, 2.1.2, 2.3.1, 2.4.0, **2.5.1** | | pytorch-engine:0.30.0 | 1.13.1, 2.1.2, 2.3.1, **2.4.0** | | pytorch-engine:0.29.0 | 1.13.1, 2.1.2, 2.2.2, **2.3.1** |