Sid Sahai a54cfe6828 Add LMSDiscreteSchedulerTest (#467) ...

* [WIP] add LMSDiscreteSchedulerTest

* fixes for comments

* add torch numpy test

* rebase

* Update tests/test_scheduler.py

* Update tests/test_scheduler.py

* style

* return residuals

Co-authored-by: Anton Lozhkov <anton@huggingface.co>

2022-09-16 19:10:56 +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

[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

[UNet2DConditionModel, UNet2DModel] pass norm_num_groups to all the blocks (#442)

2022-09-15 16:35:14 +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

Quick fix for the img2img tests (#530)

2022-09-16 15:52:26 +02:00

test_scheduler.py

Add LMSDiscreteSchedulerTest (#467)

2022-09-16 19:10:56 +02:00

test_training.py

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

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