• Victor Schmidt's avatar
    Handle batches and Tensors in... · 3bec90ff
    Victor Schmidt authored
    
    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: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    3bec90ff
test_stable_diffusion_inpaint.py 25.6 KB