diffusers/tests
Patrick von Platen a80480f0f2
[Tests] Improve unet / vae tests (#1018)
* improve tests

* up

* finish

* upload

* add init

* up

* finish vae

* finish

* reduce loading time with device_map

* remove device_map from CPU

* uP
2022-10-28 13:43:26 +02:00
..
fixtures/custom_pipeline Custome Pipelines (#744) 2022-10-06 16:54:02 +02:00
models [Tests] Improve unet / vae tests (#1018) 2022-10-28 13:43:26 +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_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