PR #942 Do not use torch.float64 on the mps device

Do not use torch.float64 on the mps device #942

pcuenca merged 6 commits into main from fix-lms-mps

Add failing test for #940.

2986f8c2

Do not use torch.float64 in mps.

e6ee09d1

pcuenca requested a review from

patrickvonplaten 3 years ago

pcuenca requested a review from

anton-l 3 years ago

pcuenca requested a review from

patil-suraj 3 years ago

style

445a6e0d

Merge remote-tracking branch 'origin/main' into fix-lms-mps

dc936fa6

patrickvonplaten approved these changes on 2022-10-25

patil-suraj approved these changes on 2022-10-25

anton-l approved these changes on 2022-10-25

Merge remote-tracking branch 'origin/main' into fix-lms-mps

065d1d85

Temporarily skip add_noise for IPNDMScheduler.

56210ad8

pcuenca merged 0343d8f5 into main 3 years ago

pcuenca deleted the fix-lms-mps branch 3 years ago

pcuenca restored the head branch 3 years ago

Reviewers

anton-l

patil-suraj

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone