lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-09 09:29:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update rng.py

Browse Source
This commit is contained in:
lllyasviel
2024-02-05 13:33:27 -08:00
parent 0ba407fd9c
commit f5b3fcc6cf
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modules/rng.py
View File

@@ -10,6 +10,11 @@ def randn(seed, shape, generator=None):
manual_seed(seed)
if generator is not None:
# if generator is not none, we must generate a noise with and without
# generator together to avoid future 'randn_like' get same noise again
torch.randn(shape, device=devices.device)
if shared.opts.randn_source == "NV":
return torch.asarray((generator or nv_rng).randn(shape), device=devices.device)