diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-03 00:00:23 +0300 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-03 00:00:23 +0300 |
| commit | 84b6fcd02ca6d6ab48c4b6be4bb8724b1c2e7014 (patch) | |
| tree | 1be8f712b6f87a29e34204acc54c0b05bc06dfb8 /modules/sd_samplers_kdiffusion.py | |
| parent | ccb92339348f6973de39cde062982a51a4cd0818 (diff) | |
add NV option for Random number generator source setting, which allows to generate same pictures on CPU/AMD/Mac as on NVidia videocards.
Diffstat (limited to 'modules/sd_samplers_kdiffusion.py')
| -rw-r--r-- | modules/sd_samplers_kdiffusion.py | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py index e0da3425..d72c1b5f 100644 --- a/modules/sd_samplers_kdiffusion.py +++ b/modules/sd_samplers_kdiffusion.py @@ -260,10 +260,7 @@ class TorchHijack: if noise.shape == x.shape:
return noise
- if opts.randn_source == "CPU" or x.device.type == 'mps':
- return torch.randn_like(x, device=devices.cpu).to(x.device)
- else:
- return torch.randn_like(x)
+ return devices.randn_like(x)
class KDiffusionSampler:
|