aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_samplers_kdiffusion.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-04-29 08:29:37 +0000
committerAUTOMATIC <16777216c@gmail.com>2023-04-29 08:29:37 +0000
commit5fe0dd79beaa5ef737ff85254ee9870f60ae9464 (patch)
treefb712561ac65b99285ad6a48b27a7920d18dde01 /modules/sd_samplers_kdiffusion.py
parentcb9571e37fc1c0ee22c92ee726147f64ee80f07a (diff)
downloadstable-diffusion-webui-gfx803-5fe0dd79beaa5ef737ff85254ee9870f60ae9464.tar.gz
stable-diffusion-webui-gfx803-5fe0dd79beaa5ef737ff85254ee9870f60ae9464.tar.bz2
stable-diffusion-webui-gfx803-5fe0dd79beaa5ef737ff85254ee9870f60ae9464.zip
rename CPU RNG to RNG source in settings, add infotext and parameters copypaste support to RNG source
Diffstat (limited to 'modules/sd_samplers_kdiffusion.py')
-rw-r--r--modules/sd_samplers_kdiffusion.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py
index 13f4567a..a547d1b5 100644
--- a/modules/sd_samplers_kdiffusion.py
+++ b/modules/sd_samplers_kdiffusion.py
@@ -190,7 +190,7 @@ class TorchHijack:
if noise.shape == x.shape:
return noise
- if opts.use_cpu_randn or x.device.type == 'mps':
+ 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)