diffusers/tests
Patrick von Platen e713346ad1
Give more customizable options for safety checker (#815)
* Give more customizable options for safety checker

* Apply suggestions from code review

* Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py

* Finish

* make style

* Apply suggestions from code review

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* up

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
2022-10-13 15:52:26 +02:00
..
fixtures/custom_pipeline Custome Pipelines (#744) 2022-10-06 16:54:02 +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 [Utils] Add deprecate function and move testing_utils under utils (#659) 2022-10-03 23:44:24 +02:00
test_modeling_common.py [Utils] Add deprecate function and move testing_utils under utils (#659) 2022-10-03 23:44:24 +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.py Fix gradient checkpointing test (#797) 2022-10-10 19:40:33 +02:00
test_models_vae.py [Utils] Add deprecate function and move testing_utils under utils (#659) 2022-10-03 23:44:24 +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 Give more customizable options for safety checker (#815) 2022-10-13 15:52:26 +02:00
test_pipelines_flax.py [Flax] Add test (#824) 2022-10-13 13:55:39 +02:00
test_scheduler.py [Scheduler design] The pragmatic approach (#719) 2022-10-05 14:41:19 +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