hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-08 13:19:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

first attempt to produce crrect seeds in batch

Browse Source
This commit is contained in:
AUTOMATIC
2022-09-13 21:49:58 +03:00
parent 85b97cc49c
commit 9d40212485
3 changed files with 51 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
modules/devices.py
View File

@@ -48,3 +48,13 @@ def randn(seed, shape):
torch.manual_seed(seed)
return torch.randn(shape, device=device)
def randn_without_seed(shape):
# Pytorch currently doesn't handle setting randomness correctly when the metal backend is used.
if device.type == 'mps':
generator = torch.Generator(device=cpu)
noise = torch.randn(shape, generator=generator, device=cpu).to(device)
return noise
return torch.randn(shape, device=device)