Suraj Patil d144c46a59 [UNet2DConditionModel, UNet2DModel] pass norm_num_groups to all the blocks (#442) ... * pass norm_num_groups to unet blocs and attention * fix UNet2DConditionModel * add norm_num_groups arg in vae * add tests * remove comment * Apply suggestions from code review		2022-09-15 16:35:14 +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	Revive `make quality` (#203)	2022-08-17 15:22:04 +02:00
test_modeling_common.py	[UNet2DConditionModel, UNet2DModel] pass norm_num_groups to all the blocks (#442)	2022-09-15 16:35:14 +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	update expected results of slow tests (#268)	2022-09-12 15:49:39 +02:00
test_scheduler.py	Karras VE, DDIM and DDPM flax schedulers (#508)	2022-09-15 15:55:48 +02:00
test_training.py	Karras VE, DDIM and DDPM flax schedulers (#508)	2022-09-15 15:55:48 +02:00