diffusers/tests/pipelines
Suraj Patil 5383188c7e
StableDiffusionDepth2ImgPipeline (#1531)
* begin depth pipeline

* add depth estimation model

* fix prepare_depth_mask

* add a comment about autocast

* copied from, quality, cleanup

* begin tests

* handle tensors

* norm image tensor

* fix batch size

* fix tests

* fix enable_sequential_cpu_offload

* fix save load

* fix test_save_load_float16

* fix test_save_load_optional_components

* fix test_float16_inference

* fix test_cpu_offload_forward_pass

* fix test_dict_tuple_outputs_equivalent

* up

* fix fast tests

* fix test_stable_diffusion_img2img_multiple_init_images

* fix few more fast tests

* don't use device map for DPT

* fix test_stable_diffusion_pipeline_with_sequential_cpu_offloading

* accept external depth maps

* prepare_depth_mask -> prepare_depth_map

* fix file name

* fix file name

* quality

* check transformers version

* fix test names

* use skipif

* fix import

* add docs

* skip tests on mps

* correct version

* uP

* Update docs/source/api/pipelines/stable_diffusion_2.mdx

* fix fix-copies

* fix fix-copies

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: anton- <anton@huggingface.co>
2022-12-08 18:25:12 +01:00
..
altdiffusion Standardize fast pipeline tests with PipelineTestMixin (#1526) 2022-12-06 18:35:30 +01:00
audio_diffusion Standardize fast pipeline tests with PipelineTestMixin (#1526) 2022-12-06 18:35:30 +01:00
dance_diffusion Standardize fast pipeline tests with PipelineTestMixin (#1526) 2022-12-06 18:35:30 +01:00
ddim Fix common tests for FP16 (#1588) 2022-12-07 14:09:51 +01:00
ddpm Standardize fast pipeline tests with PipelineTestMixin (#1526) 2022-12-06 18:35:30 +01:00
karras_ve Standardize fast pipeline tests with PipelineTestMixin (#1526) 2022-12-06 18:35:30 +01:00
latent_diffusion Standardize fast pipeline tests with PipelineTestMixin (#1526) 2022-12-06 18:35:30 +01:00
paint_by_example Add paint by example (#1533) 2022-12-07 11:06:30 +01:00
pndm Standardize fast pipeline tests with PipelineTestMixin (#1526) 2022-12-06 18:35:30 +01:00
repaint Update scheduling_repaint.py (#1582) 2022-12-07 17:41:07 +01:00
score_sde_ve Standardize fast pipeline tests with PipelineTestMixin (#1526) 2022-12-06 18:35:30 +01:00
stable_diffusion [K Diffusion] Add k diffusion sampler natively (#1603) 2022-12-08 12:48:37 +01:00
stable_diffusion_2 StableDiffusionDepth2ImgPipeline (#1531) 2022-12-08 18:25:12 +01:00
stable_diffusion_safe Standardize fast pipeline tests with PipelineTestMixin (#1526) 2022-12-06 18:35:30 +01:00
versatile_diffusion Standardize fast pipeline tests with PipelineTestMixin (#1526) 2022-12-06 18:35:30 +01:00
vq_diffusion Standardize fast pipeline tests with PipelineTestMixin (#1526) 2022-12-06 18:35:30 +01:00
__init__.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00