3bec90ff2c
* 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> |
||
|---|---|---|
| .. | ||
| altdiffusion | ||
| dance_diffusion | ||
| ddim | ||
| ddpm | ||
| karras_ve | ||
| latent_diffusion | ||
| pndm | ||
| repaint | ||
| score_sde_ve | ||
| stable_diffusion | ||
| vq_diffusion | ||
| __init__.py | ||