brkirch 0fddb4a1c0 Rework MPS randn fix, add randn_like fix ...

torch.manual_seed() already sets a CPU generator, so there is no reason to create a CPU generator manually. torch.randn_like also needs a MPS fix for k-diffusion, but a torch hijack with randn_like already exists so it can also be used for that.

2022-11-30 10:33:42 -05:00

3.2 KiB

Raw Blame History

View Raw