CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-06-04 21:05:48 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix for XPU (#1997)

Browse Source 
use float32 for XPU, same as previous fix for MPS
This commit is contained in:
DenOfEquity 2024-10-06 14:33:47 +01:00 committed by GitHub
parent 4f7f815b9f
commit 2467c88c50
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/nn/flux.py
View File

@ -19,7 +19,7 @@ def attention(q, k, v, pe):
def rope(pos, dim, theta):
if pos.device.type == "mps":
if pos.device.type == "mps" or pos.device.type == "xpu":
scale = torch.arange(0, dim, 2, dtype=torch.float32, device=pos.device) / dim
else:
scale = torch.arange(0, dim, 2, dtype=torch.float64, device=pos.device) / dim