Allow Intel GPUs to LoRA cast on GPU since it supports BF16 natively.

1 year ago · eec449ca8e
parent 29ccf9f471
commit eec449ca8e
1 changed files with 2 additions and 0 deletions
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@ -488,6 +488,8 @@ def cast_to_device(tensor, device, dtype, copy=False):
    elif tensor.dtype == torch.bfloat16:
        if hasattr(device, 'type') and device.type.startswith("cuda"):
            device_supports_cast = True
+        elif is_intel_xpu():
+            device_supports_cast = True

    if device_supports_cast:
        if copy: