From 8508a5a853672520fff81505268c21c5bbe87db5 Mon Sep 17 00:00:00 2001
From: Benjamin Berman <ben@hiddenswitch.com>
Date: Thu, 8 Feb 2024 08:40:13 -0800
Subject: [PATCH] Fix args.directml is not None error

---
 comfy/model_management.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index 20d3ed705..0325cc48b 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -37,7 +37,7 @@ if args.deterministic:
     torch.use_deterministic_algorithms(True, warn_only=True)
 
 directml_enabled = False
-if args.directml is not None:
+if args.directml:
     import torch_directml
     directml_enabled = True
     device_index = args.directml