From ebfcf0a9c91da34feec1d3ee6edd41a0302a637c Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Fri, 3 Mar 2023 13:18:01 -0500
Subject: [PATCH] Fix issue.

---
 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 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