search

divamgupta / stable-diffusion-tensorflow

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

Inpainting support and bug fix #45

Closed ShaunXZ closed 1 year ago

ShaunXZ commented 1 year ago

New features:

  1. Inpainting when both input_image and input_mask are provided.
  2. Mix precision for image2image and inpainting.

Bug fix:

  1. In StableDiffusion.get_starting_parameters(), noise should be added after tf.repeat(). Otherwise the same noise would be used for the whole batch.
  2. Fixed tensor type error when doing mixed precision. StableDiffusion now checks global policy dtype during initialization.
divamgupta commented 1 year ago

Thanks for the PR. Could you remove the changes in the README from your PR

ShaunXZ commented 1 year ago

Thanks for the PR. Could you remove the changes in the README from your PR

Sure. changes to README are now removed.