Fix Flux FP64 math on XPU (comfyanonymous#4210)

NeoWorldTeam · Aug 5, 2024 · a178e25 · a178e25
1 parent 78e133d
commit a178e25
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/comfy/ldm/flux/math.py b/comfy/ldm/flux/math.py
@@ -14,7 +14,7 @@ def attention(q: Tensor, k: Tensor, v: Tensor, pe: Tensor) -> Tensor:
 
 def rope(pos: Tensor, dim: int, theta: int) -> Tensor:
     assert dim % 2 == 0
-    if comfy.model_management.is_device_mps(pos.device):
+    if comfy.model_management.is_device_mps(pos.device) or comfy.model_management.is_intel_xpu():
         device = torch.device("cpu")
     else:
         device = pos.device