Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Pedro Cuenca's avatar
    MPS schedulers: don't use float64 (#1169) · 813744e5
    Pedro Cuenca authored 
    * Schedulers: don't use float64 on mps

* Test set_timesteps() on device (float schedulers).

* SD pipeline: use device in set_timesteps.

* SD in-painting pipeline: use device in set_timesteps.

* Tests: fix mps crashes.

* Skip test_load_pipeline_from_git on mps.

Not compatible with float16.

* Use device.type instead of str in Euler schedulers.
    813744e5
scheduling_lms_discrete.py 11.4 KB