Small cleanups.

0d7b0a4d · comfyanonymous · 92254659 · 0d7b0a4d
Commit 0d7b0a4d authored Aug 20, 2023 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

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

No files found.
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -589,7 +589,10 @@ def should_use_fp16(device=None, model_params=0):
    if cpu_mode() or mps_mode():
        return False #TODO ?

-    if torch.cuda.is_bf16_supported() or xpu_available:
+    if xpu_available:
+        return True
+
+    if torch.cuda.is_bf16_supported():
        return True

    props = torch.cuda.get_device_properties("cuda")