From e20f878f410e4ba872196d18d5afdc13e657868a Mon Sep 17 00:00:00 2001
From: Mat Sz <contact@matsz.dev>
Date: Mon, 11 Nov 2024 18:40:35 +0100
Subject: [PATCH] fix: update directml requirements

---
 .github/workflows/build.yml | 10 +++++-----
 requirements.directml.txt   |  1 +
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index b9b9d88..e326570 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -23,7 +23,7 @@ jobs:
     runs-on: ubuntu-latest
     env:
       artifact_name: linux-${{ matrix.build.arch }}-${{ matrix.build.type }}.tar.br
-      python_url: 'https://github.com/indygreg/python-build-standalone/releases/download/20241002/cpython-3.11.10+20241002-x86_64_v3-unknown-linux-gnu-install_only_stripped.tar.gz'
+      python_url: 'https://github.com/indygreg/python-build-standalone/releases/download/20241016/cpython-3.11.10+20241016-x86_64_v3-unknown-linux-gnu-install_only_stripped.tar.gz'
       pip_args: ${{ matrix.build.pip_args }} -r requirements.${{ matrix.build.type }}.txt
     steps:
       - name: Remove unnecessary files
@@ -107,14 +107,14 @@ jobs:
             pip_args: ''
           - arch: 'x64'
             type: 'directml'
-            pip_args: '--pre'
+            pip_args: ''
           - arch: 'x64'
             type: 'cpu'
             pip_args: ''
     runs-on: windows-latest
     env:
       artifact_name: win32-${{ matrix.build.arch }}-${{ matrix.build.type }}.tar.br
-      python_url: 'https://github.com/indygreg/python-build-standalone/releases/download/20241002/cpython-3.10.15+20241002-x86_64-pc-windows-msvc-shared-install_only_stripped.tar.gz'
+      python_url: 'https://github.com/indygreg/python-build-standalone/releases/download/20241016/cpython-3.11.10+20241016-x86_64-pc-windows-msvc-shared-install_only_stripped.tar.gz'
       pip_args: ${{ matrix.build.pip_args }} -r requirements.${{ matrix.build.type }}.txt
     steps:
       - uses: actions/checkout@v4
@@ -173,7 +173,7 @@ jobs:
     runs-on: macos-14
     env:
       artifact_name: darwin-${{ matrix.build.arch }}-${{ matrix.build.type }}.tar.br
-      python_url: 'https://github.com/indygreg/python-build-standalone/releases/download/20241002/cpython-3.11.10+20241002-aarch64-apple-darwin-install_only_stripped.tar.gz'
+      python_url: 'https://github.com/indygreg/python-build-standalone/releases/download/20241016/cpython-3.11.10+20241016-aarch64-apple-darwin-install_only_stripped.tar.gz'
       pip_args: -r requirements.${{ matrix.build.type }}.txt
     steps:
       - uses: actions/checkout@v4
@@ -231,7 +231,7 @@ jobs:
     runs-on: macos-13
     env:
       artifact_name: darwin-${{ matrix.build.arch }}-${{ matrix.build.type }}.tar.br
-      python_url: 'https://github.com/indygreg/python-build-standalone/releases/download/20241002/cpython-3.11.10+20241002-x86_64-apple-darwin-install_only_stripped.tar.gz'
+      python_url: 'https://github.com/indygreg/python-build-standalone/releases/download/20241016/cpython-3.11.10+20241016-x86_64-apple-darwin-install_only_stripped.tar.gz'
       pip_args: -r requirements.${{ matrix.build.type }}.txt
     steps:
       - uses: actions/checkout@v4
diff --git a/requirements.directml.txt b/requirements.directml.txt
index 7c1442b..2d601c8 100644
--- a/requirements.directml.txt
+++ b/requirements.directml.txt
@@ -1,2 +1,3 @@
 -r requirements.txt
+torch==2.4.1
 torch-directml
\ No newline at end of file