use stable version of pytorch and reenable directml

mat-sz · mat-sz · commit 33d6faa48d2c · 2024-09-18T18:56:27.000+02:00
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -16,7 +16,7 @@ jobs:
             pip_args: ''
           - arch: 'x64'
             type: 'cpu'
-            pip_args: '--pre'
+            pip_args: '--extra-index-url https://download.pytorch.org/whl/cpu'
           - arch: 'x64'
             type: 'rocm'
             pip_args: '--pre'
@@ -75,7 +75,7 @@ jobs:
           #   pip_args: '--pre'
           - arch: 'x64'
             type: 'cpu'
-            pip_args: '--pre'
+            pip_args: ''
     runs-on: windows-latest
     env:
       artifact_name: win32-${{ matrix.build.arch }}-${{ matrix.build.type }}.tar.zst
diff --git a/requirements.cpu.txt b/requirements.cpu.txt
@@ -1,3 +1,2 @@
---extra-index-url https://download.pytorch.org/whl/nightly/cpu
 -r requirements.txt
 torch
diff --git a/requirements.cuda.txt b/requirements.cuda.txt
@@ -1,4 +1,4 @@
---extra-index-url https://download.pytorch.org/whl/cu121
+--extra-index-url https://download.pytorch.org/whl/cu124
 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/
 -r requirements.txt
 torch
diff --git a/requirements.rocm.txt b/requirements.rocm.txt
@@ -1,3 +1,3 @@
---extra-index-url https://download.pytorch.org/whl/nightly/rocm6.0
+--extra-index-url https://download.pytorch.org/whl/rocm6.1
 -r requirements.txt
 torch
diff --git a/requirements.txt b/requirements.txt
@@ -1,13 +1,13 @@
 torchvision
 torchsde
 einops
-safetensors
+safetensors>=0.4.2
 pyyaml
 Pillow
 scipy
 tqdm
 psutil
-transformers
+transformers>=4.28.1
 segment_anything
 yapf
 addict

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,2 @@`
`1`		`---extra-index-url https://download.pytorch.org/whl/nightly/cpu`
`2`	`1`	`-r requirements.txt`
`3`	`2`	`torch`