Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
diffusers/tests
History
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
..
fixtures/custom_pipeline [Custom pipeline] Easier loading of local pipelines (#1327) 2022-11-17 16:00:26 +01:00
models Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01:00
pipelines Handle batches and Tensors in `pipeline_stable_diffusion_inpaint.py:prepare_mask_and_masked_image` (#1003) 2022-11-20 19:33:09 +01:00
repo_utils Move accelerate to a soft-dependency (#1134) 2022-11-04 14:58:52 +01: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 Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01:00
test_layers_utils.py VQ-diffusion (#658) 2022-11-03 16:10:28 +01:00
test_modeling_common.py Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01: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 [Custom pipeline] Easier loading of local pipelines (#1327) 2022-11-17 16:00:26 +01:00
test_pipelines_common.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_pipelines_flax.py Flax tests: don't hardcode number of devices (#1175) 2022-11-09 20:04:43 +01:00
test_pipelines_onnx_common.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_scheduler.py Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01:00
test_scheduler_flax.py Add Scheduler.from_pretrained and better scheduler changing (#1286) 2022-11-15 18:15:13 +01: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