diff --git a/gradio_demo/app-multicontrolnet.py b/gradio_demo/app-multicontrolnet.py index 2a07d7f..9269dd7 100644 --- a/gradio_demo/app-multicontrolnet.py +++ b/gradio_demo/app-multicontrolnet.py @@ -205,6 +205,7 @@ def main(pretrained_model_name_or_path="wangqixun/YamerMIX_v8", enable_lcm_arg=F 42, # seed "EulerDiscreteScheduler", # scheduler False, # enable_LCM + True, # enable_Face_Region ) def convert_from_cv2_to_image(img: np.ndarray) -> Image: diff --git a/infer_full.py b/infer_full.py index 98dbe97..d0a1bb9 100644 --- a/infer_full.py +++ b/infer_full.py @@ -98,7 +98,6 @@ if __name__ == "__main__": processed_image_midas = processed_image_midas.resize(pose_image.size) # enhance face region - width, height = face_kps.size control_mask = np.zeros([height, width, 3]) x1, y1, x2, y2 = face_info["bbox"] x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2) diff --git a/pipeline_stable_diffusion_xl_instantid_full.py b/pipeline_stable_diffusion_xl_instantid_full.py index 8af7a33..920c72b 100644 --- a/pipeline_stable_diffusion_xl_instantid_full.py +++ b/pipeline_stable_diffusion_xl_instantid_full.py @@ -1098,7 +1098,7 @@ class StableDiffusionXLInstantIDPipeline(StableDiffusionXLControlNetPipeline): return_dict=False) # controlnet mask - if control_mask_wight_image_list is not None: + if control_index == 0 and control_mask_wight_image_list is not None: down_block_res_samples = [ down_block_res_sample * mask_weight for down_block_res_sample, mask_weight in zip(down_block_res_samples, control_mask_wight_image_list)