Nathan Lambert b56f102765 Fix scheduler inference steps error with power of 3 (#466) ...

* initial attempt at solving

* fix pndm power of 3 inference_step

* add power of 3 test

* fix index in pndm test, remove ddim test

* add comments, change to round()

2022-09-13 09:48:33 -06: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

update expected results of slow tests (#268)

2022-09-12 15:49:39 +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

Fix scheduler inference steps error with power of 3 (#466)

2022-09-13 09:48:33 -06:00

test_training.py

[ModelOutputs] Replace dict outputs with Dict/Dataclass and allow to return tuples (#334)

2022-09-05 14:49:26 +02:00