Fix issue.

2 years ago · ebfcf0a9c9
parent 8586515886
commit ebfcf0a9c9
1 changed files with 1 additions and 1 deletions
--- 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