Fix issue.

ebfcf0a9 · comfyanonymous · 85865158 · ebfcf0a9
Commit ebfcf0a9 authored Mar 03, 2023 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

comfy/model_management.py comfy/model_management.py +1 -1

No files found.
--- 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