Pedro Cuenca af279434d0 Flax tests: don't hardcode number of devices (#1175) ... Flax tests: don't hardcode number of devices. This makes it possible to test on CPU/GPU. However, expected slices are only checked when there are 8 devices.		2022-11-09 20:04:43 +01:00
..
fixtures/custom_pipeline	[Scheduler] Move predict epsilon to init (#1155)	2022-11-08 18:08:08 +01:00
models	MPS schedulers: don't use float64 (#1169)	2022-11-08 13:11:33 +01:00
pipelines	Add LDM Super Resolution pipeline (#1116)	2022-11-09 13:42:16 +01:00
repo_utils	Move accelerate to a soft-dependency (#1134)	2022-11-04 14:58:52 +01:00
__init__.py	fix issues with loading, add test for pipeline	2022-06-07 15:40:36 +02:00
conftest.py	[Utils] Add deprecate function and move testing_utils under utils (#659)	2022-10-03 23:44:24 +02:00
test_config.py	[Scheduler] Move predict epsilon to init (#1155)	2022-11-08 18:08:08 +01:00
test_layers_utils.py	VQ-diffusion (#658)	2022-11-03 16:10:28 +01:00
test_modeling_common.py	Add Apple M1 tests (#796)	2022-10-17 20:27:30 +02:00
test_modeling_common_flax.py	Flax: Trickle down `norm_num_groups` (#789)	2022-10-11 20:05:10 +02:00
test_outputs.py	Fix BaseOutput initialization from dict (#570)	2022-09-20 18:32:16 +02:00
test_pipelines.py	[Loading] Make sure loading edge cases work (#1192)	2022-11-09 12:28:56 +01:00
test_pipelines_common.py	Reorganize pipeline tests (#963)	2022-10-24 16:34:01 +02:00
test_pipelines_flax.py	Flax tests: don't hardcode number of devices (#1175)	2022-11-09 20:04:43 +01:00
test_pipelines_onnx_common.py	Reorganize pipeline tests (#963)	2022-10-24 16:34:01 +02:00
test_scheduler.py	Fix slow tests (#1210)	2022-11-09 11:22:12 +01:00
test_scheduler_flax.py	[CI] Framework and hardware-specific CI tests (#997)	2022-11-02 14:07:07 +01:00
test_training.py	[Utils] Add deprecate function and move testing_utils under utils (#659)	2022-10-03 23:44:24 +02:00
test_utils.py	[Utils] Add deprecate function and move testing_utils under utils (#659)	2022-10-03 23:44:24 +02:00