Nouamane Tazi b64c522759 [PNDM Scheduler] format timesteps attrs to np arrays (#273) ...

* format timesteps attrs to np arrays in pndm scheduler
because lists don't get formatted to tensors in `self.set_format`

* convert to long type to use timesteps as indices for tensors

* add scheduler set_format test

* fix `_timesteps` type

* make style with black 22.3.0 and isort 5.10.1

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

2022-08-31 14:12:08 +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

split tests_modeling_utils (#223)

2022-08-24 13:27:16 +02:00

test_models_unet.py

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

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

test_models_vae.py

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

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

test_models_vq.py

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

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

test_pipelines.py

Refactor Pipelines / Community pipelines and add better explanations. (#257)

2022-08-30 18:43:42 +02:00

test_scheduler.py

[PNDM Scheduler] format timesteps attrs to np arrays (#273)

2022-08-31 14:12:08 +02:00

test_training.py

Add training and batched inference test for DDPM vs DDIM (#140)

2022-07-27 15:01:56 +02:00