From f7a5107784cded39f92a4bb7553507575e78edbe Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Sat, 3 Aug 2024 16:55:38 -0400
Subject: [PATCH] Fix crash.

---
 comfy/model_management.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index b280b14..fb27470 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -928,10 +928,7 @@ def should_use_bf16(device=None, model_params=0, prioritize_performance=True, ma
     if is_intel_xpu():
         return True
 
-    if device is None:
-        device = torch.device("cuda")
-
-    props = torch.cuda.get_device_properties(device)
+    props = torch.cuda.get_device_properties("cuda")
     if props.major >= 8:
         return True