diffusers/docs/source/api
Kashif Rasul bd8df2da89
[Pytorch] Pytorch only schedulers (#534)
* pytorch only schedulers

* fix style

* remove match_shape

* pytorch only ddpm

* remove SchedulerMixin

* remove numpy from karras_ve

* fix types

* remove numpy from lms_discrete

* remove numpy from pndm

* fix typo

* remove mixin and numpy from sde_vp and ve

* remove remaining tensor_format

* fix style

* sigmas has to be torch tensor

* removed set_format in readme

* remove set format from docs

* remove set_format from pipelines

* update tests

* fix typo

* continue to use mixin

* fix imports

* removed unsed imports

* match shape instead of assuming image shapes

* remove import typo

* update call to add_noise

* use math instead of numpy

* fix t_index

* removed commented out numpy tests

* timesteps needs to be discrete

* cast timesteps to int in flax scheduler too

* fix device mismatch issue

* small fix

* Update src/diffusers/schedulers/scheduling_pndm.py

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-09-27 15:27:34 +02:00
..
pipelines docs: `.md` readability fixups (#619) 2022-09-23 12:02:27 +02:00
configuration.mdx Add config docs (#429) 2022-09-08 17:46:03 +02:00
diffusion_pipeline.mdx [Outputs] Improve syntax (#423) 2022-09-08 16:46:38 +02:00
logging.mdx [Docs] Logging (#394) 2022-09-07 14:58:21 +02:00
models.mdx Flax documentation (#589) 2022-09-23 13:24:16 +02:00
outputs.mdx [Docs] Outputs.mdx (#422) 2022-09-08 14:47:14 +02:00
schedulers.mdx [Pytorch] Pytorch only schedulers (#534) 2022-09-27 15:27:34 +02:00