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

maybe fix slerp

Browse Source
This commit is contained in:
Vladimir Mandic
2024-04-03 17:51:00 -04:00
parent ce2bb713d8
commit 40d5fdfdfd
5 changed files with 24 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/devices.py
+4 -1
View File
@@ -369,7 +369,10 @@ def randn(seed, shape):
torch.xpu.manual_seed_all(seed)
if device.type == 'mps':
return torch.randn(shape, device=cpu).to(device)
return torch.randn(shape, device=device)
elif shared.opts.diffusers_generator_device == "CPU":
return torch.randn(shape, device=cpu)
else:
return torch.randn(shape, device=device)
def randn_without_seed(shape):