diff --git a/comfy/cli_args.py b/comfy/cli_args.py
index e8e62c3d0..3f66fb856 100644
--- a/comfy/cli_args.py
+++ b/comfy/cli_args.py
@@ -119,6 +119,7 @@ parser.add_argument("--plausible-analytics-domain", required=False,
                     help="Specifies the domain name for analytics events.")
 parser.add_argument("--analytics-use-identity-provider", action="store_true",
                     help="Uses platform identifiers for unique visitor analytics.")
+
 if options.args_parsing:
     args, _ = parser.parse_known_args()
 else:
diff --git a/comfy/model_management.py b/comfy/model_management.py
index 0325cc48b..20d3ed705 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:
+if args.directml is not None:
     import torch_directml
     directml_enabled = True
     device_index = args.directml