add torch full deterministic mode

This commit is contained in:
Vladimir Mandic
2024-06-07 09:26:51 -04:00
parent 5574833f0d
commit db9718eee6
9 changed files with 61 additions and 51 deletions
+3
View File
@@ -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