From 98d0bd2d4a9d218344b11fa0e1547857b4b99efd Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Tue, 26 Dec 2023 19:38:36 +0100
Subject: [PATCH] Hotfix onnxruntime-directml for Windows

---
 facefusion/installer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/facefusion/installer.py b/facefusion/installer.py
index 8b6f7fe..1b1d563 100644
--- a/facefusion/installer.py
+++ b/facefusion/installer.py
@@ -29,11 +29,12 @@ if platform.system().lower() == 'linux' or platform.system().lower() == 'windows
 	ONNXRUNTIMES['openvino'] = ('onnxruntime-openvino', '1.16.0')
 if platform.system().lower() == 'linux':
 	TORCH['rocm'] = 'rocm5.6'
-	ONNXRUNTIMES['directml'] = ('onnxruntime-directml', '1.16.3')
 	ONNXRUNTIMES['rocm'] = ('onnxruntime-rocm', '1.16.3')
 if platform.system().lower() == 'darwin':
 	ONNXRUNTIMES['coreml-legacy'] = ('onnxruntime-coreml', '1.13.1')
 	ONNXRUNTIMES['coreml-silicon'] = ('onnxruntime-silicon', '1.16.0')
+if platform.system().lower() == 'windows':
+	ONNXRUNTIMES['directml'] = ('onnxruntime-directml', '1.16.3')
 
 
 def cli() -> None: