diffusers/tests/pipelines/stable_diffusion
Victor Schmidt 3bec90ff2c
Handle batches and Tensors in `pipeline_stable_diffusion_inpaint.py:prepare_mask_and_masked_image` (#1003)
* Handle batches and Tensors in `prepare_mask_and_masked_image`

* `blackfy`
upgrade `black`

* handle mask as `np.array`

* add docstring

* revert `black` changes with smaller line length

* missing ValueError in docstring

* raise `TypeError` for image as tensor but not mask

* typo in mask shape selection

* check for batch dim

* fix: wrong indentation

* add tests

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-11-20 19:33:09 +01:00
..
__init__.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_cycle_diffusion.py Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01:00
test_onnx_stable_diffusion.py Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01:00
test_onnx_stable_diffusion_img2img.py Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01:00
test_onnx_stable_diffusion_inpaint.py Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01:00
test_onnx_stable_diffusion_inpaint_legacy.py Legacy Inpainting Pipeline for Onnx Models (#1237) 2022-11-18 16:33:12 +01:00
test_stable_diffusion.py Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01:00
test_stable_diffusion_img2img.py Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01:00
test_stable_diffusion_inpaint.py Handle batches and Tensors in `pipeline_stable_diffusion_inpaint.py:prepare_mask_and_masked_image` (#1003) 2022-11-20 19:33:09 +01:00
test_stable_diffusion_inpaint_legacy.py Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01:00