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