diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index c6d95ff..81dbfc6 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -13,16 +13,19 @@ jobs: build: - arch: 'x64' type: 'cuda' + uv_args: '' - arch: 'x64' type: 'cpu' + uv_args: '--prerelease=allow' - arch: 'x64' type: 'rocm' + uv_args: '--prerelease=allow' runs-on: ubuntu-latest env: artifact_name: linux-${{ matrix.build.arch }}-${{ matrix.build.type }}.tar.zst python_url: 'https://github.com/indygreg/python-build-standalone/releases/download/20240415/cpython-3.11.9+20240415-x86_64_v3-unknown-linux-gnu-install_only.tar.gz' uv_url: 'https://github.com/astral-sh/uv/releases/download/0.1.35/uv-x86_64-unknown-linux-gnu.tar.gz' - requirements: requirements.${{ matrix.build.type }}.txt + requirements: ${{ matrix.build.uv_args }} requirements.${{ matrix.build.type }}.txt steps: - uses: actions/checkout@v4 @@ -78,16 +81,19 @@ jobs: build: - arch: 'x64' type: 'cuda' + uv_args: '' - arch: 'x64' type: 'directml' + uv_args: '--prerelease=allow' - arch: 'x64' type: 'cpu' + uv_args: '--prerelease=allow' runs-on: windows-latest env: artifact_name: win32-${{ matrix.build.arch }}-${{ matrix.build.type }}.tar.zst python_url: 'https://github.com/indygreg/python-build-standalone/releases/download/20240415/cpython-3.11.9+20240415-x86_64-pc-windows-msvc-install_only.tar.gz' uv_url: 'https://github.com/astral-sh/uv/releases/download/0.1.35/uv-x86_64-pc-windows-msvc.zip' - requirements: requirements.${{ matrix.build.type }}.txt + requirements: ${{ matrix.build.uv_args }} requirements.${{ matrix.build.type }}.txt steps: - uses: actions/checkout@v4 diff --git a/requirements.cpu.txt b/requirements.cpu.txt index f361b1d..d775cc7 100644 --- a/requirements.cpu.txt +++ b/requirements.cpu.txt @@ -1,4 +1,3 @@ ---pre --extra-index-url https://download.pytorch.org/whl/nightly/cpu -r requirements.txt torch diff --git a/requirements.directml.txt b/requirements.directml.txt index ec53889..97cc5ca 100644 --- a/requirements.directml.txt +++ b/requirements.directml.txt @@ -1,4 +1,3 @@ ---pre -r requirements.txt torch-directml onnxruntime-directml \ No newline at end of file diff --git a/requirements.rocm.txt b/requirements.rocm.txt index 237449f..de7c9a9 100644 --- a/requirements.rocm.txt +++ b/requirements.rocm.txt @@ -1,4 +1,3 @@ ---pre --extra-index-url https://download.pytorch.org/whl/nightly/rocm6.0 -r requirements.txt torch