Skip to content

Add inpainting pipeline (SD 2.1)#13

Open
TroyHernandez wants to merge 1 commit intomainfrom
feat/inpainting-v3
Open

Add inpainting pipeline (SD 2.1)#13
TroyHernandez wants to merge 1 commit intomainfrom
feat/inpainting-v3

Conversation

@TroyHernandez
Copy link
Contributor

Summary

  • Add inpaint() and inpaint_sd21() for mask-based image editing
  • Supports both native torch and TorchScript UNet backends
  • Includes mask preprocessing utilities

Replaces #9 (rebased cleanly on main after hfhub migration merge).

🤖 Generated with Claude Code

SD 2.1 inpainting pipeline: encode image, add noise, denoise with
mask blending at each DDIM step. preprocess_mask() helper loads,
resizes, and binarizes masks to latent dimensions.

Mask convention: white=1=inpaint, black=0=keep.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant