Don't use is_bf16_supported to check for fp16 support.

Eleo-ai · Feb 5, 2024 · 66e28ef · 66e28ef
1 parent 24129d7
commit 66e28ef
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/comfy/model_management.py b/comfy/model_management.py
@@ -722,10 +722,13 @@ def should_use_fp16(device=None, model_params=0, prioritize_performance=True, ma
     if is_intel_xpu():
         return True
 
-    if torch.cuda.is_bf16_supported():
+    if torch.version.hip:
         return True
 
     props = torch.cuda.get_device_properties("cuda")
+    if props.major >= 8:
+        return True
+
     if props.major < 6:
         return False