Suraj Patil 84b9df57a7 [gradient checkpointing] lower tolerance for test (#652) ...

* lowe tolerance

* put model in eval mode

2022-09-29 11:57:37 +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

[UNet2DConditionModel] add gradient checkpointing (#461)

2022-09-22 15:36:47 +02:00

test_models_unet.py

[gradient checkpointing] lower tolerance for test (#652)

2022-09-29 11:57:37 +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

[Pytorch] Pytorch only schedulers (#534)

2022-09-27 15:27:34 +02:00

test_scheduler.py

Fix the LMS pytorch regression (#664)

2022-09-28 14:07:26 +02:00

test_training.py

[Pytorch] Pytorch only schedulers (#534)

2022-09-27 15:27:34 +02:00