From c21a10b7c9edd91c84ce62d570449d6f42612989 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sun, 29 Sep 2024 21:34:59 -0400 Subject: [PATCH] add bf16 override for directml Signed-off-by: Vladimir Mandic --- modules/devices.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/devices.py b/modules/devices.py index e14594d21..596debfee 100644 --- a/modules/devices.py +++ b/modules/devices.py @@ -272,7 +272,7 @@ def test_bf16(): global bf16_ok # pylint: disable=global-statement if bf16_ok is not None: return bf16_ok - if sys.platform == "darwin" or backend == 'openvino': # override + if sys.platform == "darwin" or backend == 'openvino' or backend == 'directml': # override bf16_ok = False return bf16_ok try: