diff --git a/comfy/model_management.py b/comfy/model_management.py
index 3bd4d27..32159b8 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -185,7 +185,7 @@ def should_use_fp16():
     if torch.cuda.is_bf16_supported():
         return True
 
-    props = torch.cuda.get_device_properties()
+    props = torch.cuda.get_device_properties("cuda")
     if props.major < 7:
         return False