Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

Remove cpu=xpu with ipex

Browse Source
This commit is contained in:
Disty0
2023-07-05 00:12:07 +03:00
parent 966eed8dd9
commit 45d50bd106
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/devices.py
+1 -7
View File
@@ -207,13 +207,7 @@ if backend == 'ipex':
args[5], args[6], args[7], args[8]).to(get_cuda_device_string()),
lambda *args, **kwargs: args[1].device != torch.device("cpu"))
#Use XPU instead of CPU. %20 Perf improvement on weak CPUs.
if args.device_id is not None:
cpu = torch.device(f"xpu:{args.device_id}")
else:
cpu = torch.device("xpu")
else:
cpu = torch.device("cpu")
cpu = torch.device("cpu")
device = device_interrogate = device_gfpgan = device_esrgan = device_codeformer = None
dtype = torch.float16
dtype_vae = torch.float16