mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 01:04:32 +02:00
add torch full deterministic mode
This commit is contained in:
@@ -482,7 +482,10 @@ def get_generator(p):
|
||||
else:
|
||||
generator_device = devices.cpu if shared.opts.diffusers_generator_device == "CPU" else shared.device
|
||||
try:
|
||||
devices.randn(p.seeds[0])
|
||||
generator = [torch.Generator(generator_device).manual_seed(s) for s in p.seeds]
|
||||
seeds = [g.initial_seed() for g in generator]
|
||||
shared.log.debug(f'Torch generator: device={generator_device} seeds={seeds}')
|
||||
except Exception as e:
|
||||
shared.log.error(f'Torch generator: seeds={p.seeds} device={generator_device} {e}')
|
||||
generator = None
|
||||
|
||||
Reference in New Issue
Block a user