VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-18 01:30:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

IPEX fix SDPA

Browse Source
This commit is contained in:
Disty0
2023-12-19 22:59:06 +03:00
parent aff05e043f
commit 8556b9d7f5
2 changed files with 64 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
library/ipex/hijacks.py
View File

@@ -185,6 +185,10 @@ def ipex_hijacks():
CondFunc('torch.Generator',
lambda orig_func, device=None: torch.xpu.Generator(return_xpu(device)),
lambda orig_func, device=None: device is not None and device != torch.device("cpu") and device != "cpu")
else:
CondFunc('torch.Generator',
lambda orig_func, device=None: orig_func(return_xpu(device)),
lambda orig_func, device=None: check_device(device))
# TiledVAE and ControlNet:
CondFunc('torch.batch_norm',