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

fix samplers

Browse Source
This commit is contained in:
Vladimir Mandic
2023-05-30 15:28:41 -04:00
parent 873371f1e5
commit fa51d45bc3
3 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_samplers_kdiffusion.py
+3 -3
View File
@@ -212,9 +212,9 @@ 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:
# 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)