Small cleanups.

2 years ago · 0d7b0a4dc7
parent 9225465975
commit 0d7b0a4dc7
1 changed files with 4 additions and 1 deletions
--- 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")