Skip to content

Commit 2db58a9

Browse files
authored
Merge pull request #43 from sail-sg/ref_localnorm
add local feature ref to local inpaint region
2 parents 55677b9 + efd66f4 commit 2db58a9

7 files changed

+210
-90
lines changed

editany_lora.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -685,6 +685,8 @@ def process(
685685
a_prompt += generated_prompt
686686
print("Generated ref text:", ref_prompt)
687687
print("Generated input text:", a_prompt)
688+
# ref_image = cropped_ref_image
689+
# ref_mask = cropped_ref_mask
688690

689691
else:
690692
ref_mask = None
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.

utils/stable_diffusion_controlnet_inpaint.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1512,6 +1512,7 @@ def __call__(
15121512
self.do_classifier_free_guidance = do_classifier_free_guidance
15131513
self.style_fidelity = style_fidelity
15141514
self.ref_mask = ref_mask
1515+
self.inpaint_mask = mask_image
15151516
attn_modules, gn_modules = self.redefine_ref_model(
15161517
self.unet, reference_attn, reference_adain, model_type="unet"
15171518
)

utils/stable_diffusion_reference.py

Lines changed: 207 additions & 90 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)