feat(pipeline): support string path for images in img2img pipeline

kjsman / stable-diffusion-pytorch

Yet another PyTorch implementation of Stable Diffusion (probably easy to read)

MIT License

540 stars 62 forks source link

feat(pipeline): support string path for images in img2img pipeline #2

Closed mspronesti closed 1 year ago

mspronesti commented 2 years ago

Hello again @kjsman, as discussed in #1, this PR aims at allowing the user to either provide a string path or a PIL.Image to guide the diffusion in the img2img pipeline. Therefore now the user can pass one of these

a list of strings
a list of PIL.Image
a list containing both

Have a look at this notebook to see what changed at user level :smile:

kjsman commented 1 year ago

Gonna merge in minutes. Thanks for your contributions! 👍