Anton Lozhkov 8a6833b85c Add the K-LMS scheduler to the inpainting pipeline + tests (#587) ...

* Add the K-LMS scheduler to the inpainting pipeline + tests

* Remove redundant casts

2022-09-20 19:10:44 +02:00

..

__init__.py

fix issues with loading, add test for pipeline

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

conftest.py

[Tests] Upload custom test artifacts (#572)

2022-09-19 17:08:29 +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

[Tests] Fix spatial transformer tests on GPU (#531)

2022-09-16 16:04:37 +02:00

test_modeling_common.py

Fix typos and add Typo check GitHub Action (#483)

2022-09-16 15:36:51 +02:00

test_models_unet.py

[Tests] Mark the ncsnpp model tests as slow (#575)

2022-09-19 17:20:58 +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_outputs.py

Fix BaseOutput initialization from dict (#570)

2022-09-20 18:32:16 +02:00

test_pipelines.py

Add the K-LMS scheduler to the inpainting pipeline + tests (#587)

2022-09-20 19:10:44 +02:00

test_scheduler.py

Unify offset configuration in DDIM and PNDM schedulers (#479)

2022-09-17 14:07:43 +02:00

test_training.py

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

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