Grigory Sizov 35e9209601 Fix formula for noise levels in Karras scheduler and tests (#627) ... fix formula for noise levels in karras scheduler and tests		2022-09-24 18:24:08 +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	[UNet2DConditionModel] add gradient checkpointing (#461)	2022-09-22 15:36:47 +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	Fix formula for noise levels in Karras scheduler and tests (#627)	2022-09-24 18:24:08 +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