• brkirch's avatar
    Rework MPS randn fix, add randn_like fix · 0fddb4a1
    brkirch authored
    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.
    0fddb4a1
sd_samplers.py 21.4 KB