Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

add manual seed

Browse Source
This commit is contained in:
Vladimir Mandic
2023-05-19 08:34:43 -04:00
parent df65e8e30a
commit 9033499e08
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extensions-builtin/sd-webui-controlnet
+1 -1
Submodule extensions-builtin/sd-webui-controlnet updated: 8d84f1f74f...1c994ff4f7
modules/devices.py
+2
View File
@@ -161,6 +161,8 @@ def cond_cast_float(tensor):
def randn(seed, shape):
torch.manual_seed(seed)
if shared.cmd_opts.use_ipex:
torch.xpu.manual_seed_all(seed)
if device.type == 'mps':
return torch.randn(shape, device=cpu).to(device)
return torch.randn(shape, device=device)