Fix crash.

7 months ago · f7a5107784
parent 91be9c2867
commit f7a5107784
1 changed files with 1 additions and 4 deletions
--- 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