Kashif Rasul b34be039f9 Karras VE, DDIM and DDPM flax schedulers (#508) ...

* beta never changes removed from state

* fix typos in docs

* removed unused var

* initial ddim flax scheduler

* import

* added dummy objects

* fix style

* fix typo

* docs

* fix typo in comment

* set return type

* added flax ddom

* fix style

* remake

* pass PRNG key as argument and split before use

* fix doc string

* use config

* added flax Karras VE scheduler

* make style

* fix dummy

* fix ndarray type annotation

* replace returns a new state

* added lms_discrete scheduler

* use self.config

* add_noise needs state

* use config

* use config

* docstring

* added flax score sde ve

* fix imports

* fix typos

2022-09-15 15:55:48 +02:00

..

__init__.py

fix issues with loading, add test for pipeline

2022-06-07 15:40:36 +02:00

test_config.py

[Tests] Make sure tests are on GPU (#269)

2022-08-29 15:58:11 +02:00

test_layers_utils.py

Revive make quality (#203)

2022-08-17 15:22:04 +02:00

test_modeling_common.py

Rename test_scheduler_outputs_equivalence in model tests. (#451)

2022-09-13 15:03:36 +02:00

test_models_unet.py

update expected results of slow tests (#268)

2022-09-12 15:49:39 +02:00

test_models_vae.py

Fix vae tests for cpu and gpu (#480)

2022-09-13 19:14:20 +02:00

test_models_vq.py

update expected results of slow tests (#268)

2022-09-12 15:49:39 +02:00

test_pipelines.py

update expected results of slow tests (#268)

2022-09-12 15:49:39 +02:00

test_scheduler.py

Karras VE, DDIM and DDPM flax schedulers (#508)

2022-09-15 15:55:48 +02:00

test_training.py

Karras VE, DDIM and DDPM flax schedulers (#508)

2022-09-15 15:55:48 +02:00