Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix xpu generator

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-06-08 10:18:12 +02:00
parent fefa1f210d
commit f27c8584ee
2 changed files with 17 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_helpers.py
+13 -10
View File
@@ -511,18 +511,21 @@ def get_generator(p):
if gen_device_opt == "Unset":
generator_device = None
generator = None
elif getattr(p, "generator", None) is not None:
generator_device = devices.cpu if gen_device_opt == "CPU" else shared.device
generator = p.generator
else:
generator_device = devices.cpu if gen_device_opt == "CPU" else shared.device
try:
p.seeds = [seed if seed != -1 else get_fixed_seed(seed) for seed in p.seeds if seed]
devices.randn(p.seeds[0])
generator = [torch.Generator(generator_device).manual_seed(s) for s in p.seeds]
except Exception as e:
log.error(f'Torch generator: seeds={p.seeds} device={generator_device} {e}')
generator = None
# Intel XPU does not support generators directly on xpu for Diffusers noise creation.
if generator_device is not None and str(generator_device).startswith("xpu"):
generator_device = devices.cpu
if getattr(p, "generator", None) is not None:
generator = p.generator
else:
try:
p.seeds = [seed if seed != -1 else get_fixed_seed(seed) for seed in p.seeds if seed]
devices.randn(p.seeds[0])
generator = [torch.Generator(generator_device).manual_seed(s) for s in p.seeds]
except Exception as e:
log.error(f'Torch generator: seeds={p.seeds} device={generator_device} {e}')
generator = None
return generator