mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-06 02:30:30 -08:00
Code Issues Projects Releases Packages Wiki Activity

add NV option for Random number generator source setting, which allows to generate same pictures on CPU/AMD/Mac as on NVidia videocards.

Browse source
This commit is contained in:
AUTOMATIC1111 2023-08-03 00:00:23 +03:00
parent ccb9233934
commit 84b6fcd02c
5 changed files with 142 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

5
modules/sd_samplers_kdiffusion.py
View file

@ -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: