Hotfix onnxruntime-directml for Windows

aboutsome · Dec 26, 2023 · 98d0bd2 · 98d0bd2
1 parent b267629
commit 98d0bd2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/facefusion/installer.py b/facefusion/installer.py
@@ -29,11 +29,12 @@
 	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: