diffusers/tests
Patrick von Platen 3be9fa97d6
[Accelerate model loading] Fix meta device and super low memory usage (#1016)
* [Accelerate model loading] Fix meta device and super low memory usage

* better naming
2022-10-27 12:11:42 +02:00
..
fixtures/custom_pipeline Custome Pipelines (#744) 2022-10-06 16:54:02 +02:00
pipelines [Accelerate model loading] Fix meta device and super low memory usage (#1016) 2022-10-27 12:11:42 +02:00
repo_utils Introduce the copy mechanism (#924) 2022-10-20 20:26:03 +02: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 [Tests] Make sure tests are on GPU (#269) 2022-08-29 15:58:11 +02:00
test_layers_utils.py Add Apple M1 tests (#796) 2022-10-17 20:27:30 +02: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_models_unet_1d.py [Dance Diffusion] Add dance diffusion (#803) 2022-10-25 18:39:25 +02:00
test_models_unet_2d.py [Dance Diffusion] Add dance diffusion (#803) 2022-10-25 18:39:25 +02:00
test_models_vae.py [MPS] fix mps failing tests (#934) 2022-10-22 09:33:40 +02:00
test_models_vae_flax.py Flax: Trickle down `norm_num_groups` (#789) 2022-10-11 20:05:10 +02:00
test_models_vq.py [Utils] Add deprecate function and move testing_utils under utils (#659) 2022-10-03 23:44:24 +02:00
test_outputs.py Fix BaseOutput initialization from dict (#570) 2022-09-20 18:32:16 +02:00
test_pipelines.py [Accelerate model loading] Fix meta device and super low memory usage (#1016) 2022-10-27 12:11:42 +02:00
test_pipelines_common.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_pipelines_flax.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_pipelines_onnx_common.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_scheduler.py Continuation of #942: additional float64 failure (#996) 2022-10-27 10:21:40 +02:00
test_scheduler_flax.py [Flax] added broadcast_to_shape_from_left helper and Scheduler tests (#864) 2022-10-25 13:43:24 +02: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