update

instantX-research · Feb 1, 2024 · 9669752 · 9669752
1 parent b3204c7
commit 9669752
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/gradio_demo/app-multicontrolnet.py b/gradio_demo/app-multicontrolnet.py
@@ -205,6 +205,7 @@ def run_for_examples(face_file, pose_file, prompt, style, negative_prompt):
             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
@@ -98,7 +98,6 @@ def resize_img(input_image, max_side=1280, min_side=1024, size=None,
     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
@@ -1098,7 +1098,7 @@ def __call__(
                                                                                   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)