diff --git a/.github/workflows/test_and_deploy.yml b/.github/workflows/test_and_deploy.yml index 0c41b4ae..a1f87e3f 100644 --- a/.github/workflows/test_and_deploy.yml +++ b/.github/workflows/test_and_deploy.yml @@ -39,6 +39,7 @@ jobs: runs-on: ${{ matrix.os }} env: PYTORCH_ENABLE_MPS_FALLBACK: "1" + PYTORCH_MPS_HIGH_WATERMARK_RATIO: "0.0" strategy: matrix: # Run all supported Python versions on linux