fix uv arguments

metastable-studio · Apr 21, 2024 · 1b7e71a · 1b7e71a
1 parent d3a0737
commit 1b7e71a
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 5 deletions.
diff --git 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
@@ -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
@@ -1,4 +1,3 @@
---pre
 -r requirements.txt
 torch-directml
 onnxruntime-directml
diff --git 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