From a178e25912b01abf436eba1cfaab316ba02d272d Mon Sep 17 00:00:00 2001
From: a-One-Fan <100067309+a-One-Fan@users.noreply.github.com>
Date: Mon, 5 Aug 2024 08:26:20 +0300
Subject: [PATCH] Fix Flux FP64 math on XPU (#4210)

---
 comfy/ldm/flux/math.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy/ldm/flux/math.py b/comfy/ldm/flux/math.py
index 88c2b6bb..136ce2aa 100644
--- 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