Method run_example

demo_lora.py:438–478

(task_type, edit_type, prompt, ref_image, edit_image, edit_mask,
                        output_h, output_w, seed, progress=gr.Progress(track_tqdm=True))

                         queue=True)

        def run_example(task_type, edit_type, prompt, ref_image, edit_image, edit_mask,
                        output_h, output_w, seed, progress=gr.Progress(track_tqdm=True)):
            model_path = self.task_model[task_type]["MODEL_PATH"]

            step = self.pipe.input.get("sample_steps", 20)
            cfg_scale = self.pipe.input.get("guide_scale", 20)

            edit_info = self.edit_type_dict[edit_type]

            edit_image = self.construct_edit_image(edit_image, edit_mask)

            pre_edit_image, pre_edit_mask, pre_ref_image, _, _ = preprocess_input(ref_image, edit_image)
            pre_edit_image = edit_preprocess(edit_info, we.device_id, pre_edit_image, pre_edit_mask)
            edit_info = edit_info or {}
            repainting_scale = edit_info.get("REPAINTING_SCALE", 1.0)
            st = time.time()
            image, seed = self.pipe(
                reference_image=pre_ref_image,
                edit_image=pre_edit_image,
                edit_mask=pre_edit_mask,
                prompt=prompt,
                output_height=output_h,
                output_width=output_w,
                sampler='flow_euler',
                sample_steps=step,
                guide_scale=cfg_scale,
                seed=seed,
                repainting_scale=repainting_scale,
                lora_path=model_path
            )
            et = time.time()
            msg = f"prompt: {prompt}; seed: {seed}; cost time: {et - st}s; repaiting scale: {repainting_scale}"
            if pre_edit_image is not None:
                ret_image = Image.composite(Image.new("RGB", pre_edit_image.size, (0, 0, 0)), pre_edit_image,  pre_edit_mask)
            else:
                ret_image = None
            return (gr.Image(value=image), gr.Column(visible=True),
                    gr.Image(value=pre_edit_image if pre_edit_image is not None else pre_ref_image),
                    gr.Image(value=pre_edit_mask if pre_edit_mask is not None else None),
                    gr.Text(value=msg),
                    gr.update(value=ret_image))

        with self.eg:
            self.example_edit_image = gr.Image(label='Edit Image',