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	…
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