From 9a5ec9c4d69c6b066415362b1dcec3f24ff4036f Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 5 Jan 2023 15:47:58 -0600 Subject: [PATCH] fix(docs): note DirectML for Windows --- README.md | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 823108a0..f2dea1d3 100644 --- a/README.md +++ b/README.md @@ -119,7 +119,12 @@ Install the following packages for AI: > pip install numpy>=1.20,<1.24 # version is important, 1.24 removed the deprecated np.float symbol > pip install accelerate diffusers transformers ftfy spacy scipy > pip install onnx onnxruntime torch -> pip install onnxruntime-directml --force-reinstall # TODO: is this one necessary? +``` + +If you are running on Windows, install the DirectML ONNX runtime as well: + +```shell +> pip install onnxruntime-directml --force-reinstall ``` Install the following packages for the web UI: