update mask start step

app.py

@@ -188,6 +188,13 @@ with gr.Blocks(css='style.css') as demo:
                                       maximum=100,
                                       step=1,
                                       value=50)
+
+                start_step = gr.Slider(label='Mask Starting Step',
+                                      minimum=0,
+                                      maximum=100,
+                                      step=1,
+                                      value=0)
+
                 guidance_scale = gr.Slider(label='CFG Scale',
                                            minimum=0,
                                            maximum=50,
@@ -222,6 +229,7 @@ with gr.Blocks(css='style.css') as demo:
                 'data/ikun/reference_images/zhongli.jpg',
                 'man',
                 0,
+                0,
                 0.5,
                 0.5,
                 0,
@@ -239,6 +247,7 @@ with gr.Blocks(css='style.css') as demo:
                 'data/huaqiang/reference_images/musk.jpg',
                 'man',
                 0,
+                0,
                 0.5,
                 0.5,
                 0,
@@ -256,6 +265,7 @@ with gr.Blocks(css='style.css') as demo:
                 'data/yanzi/reference_images/panda.jpeg',
                 'panda',
                 0,
+                0,
                 0.5,
                 0.5,
                 0,
@@ -273,6 +283,7 @@ with gr.Blocks(css='style.css') as demo:
                 'data/car-turn/reference_images/audi.jpeg',
                 'car',
                 0,
+                0,
                 0.0,
                 1.0,
                 0,
@@ -290,6 +301,7 @@ with gr.Blocks(css='style.css') as demo:
                 'data/car-turn/images/0000.jpg',
                 'car',
                 0,
+                0,
                 0.0,
                 1.0,
                 1,
@@ -309,6 +321,7 @@ with gr.Blocks(css='style.css') as demo:
                         ref_image,
                         ref_pro_prompt,
                         noise_level,
+                        start_step,
                         control_pose,
                         control_depth,
                         source_pro,
@@ -338,6 +351,7 @@ with gr.Blocks(css='style.css') as demo:
         ref_image,
         ref_pro_prompt,
         noise_level,
+        start_step,
         control_pose,
         control_depth,
         source_pro,

inference.py

@@ -204,6 +204,7 @@ class InferencePipeline:
         ref_image: PIL.Image.Image,
         ref_pro_prompt: str,
         noise_level: int,
+        start_step: int, 
         control_pose: float,
         control_depth: float,
         source_pro: int = 0, # 0 or 1
@@ -266,6 +267,7 @@ class InferencePipeline:
             masks=masks, 
             mask_mode='all',
             mask_latent_fuse_mode = 'all',
+            start_step=start_step,
             ## edit bg and pro
             prior_latents=None, 
             image_embeds=image_embed, # keep pro