divamgupta / stable-diffusion-tensorflow

Stable Diffusion in TensorFlow / Keras
Other
1.57k stars 227 forks source link

How to use Inpainting function #51

Closed a53769 closed 1 year ago

a53769 commented 1 year ago

I see, some codes had already written at stable_diffusion.py, but if use input_mask param directly It cause a error in runtime.

ShaunXZ commented 1 year ago

The input_mask param is the path to your mask file and should be used together with input_image. The mask is a single-channel uint8 array with two dimensions.

Could you try this colab example and see if that works?