diff --git a/scripts/eyemask/script.py b/scripts/eyemask/script.py index 1b06067..256f58a 100644 --- a/scripts/eyemask/script.py +++ b/scripts/eyemask/script.py @@ -212,15 +212,21 @@ class EyeMasksCore(): updated_info = self.update_info(initial_info, em_params) output_images.append(processed.images[0]) - p.all_seeds.append(start_seed) - p.all_prompts.append(save_prompt) - p.infotexts.append(updated_info) + try: + p.all_seeds.append(start_seed) + p.all_prompts.append(save_prompt) + p.infotexts.append(updated_info) + except Exception as e: + pass if em_include_mask and (n == iterations - 1 or (is_txt2img and em_redraw_original)): output_images.append(mask) - p.all_seeds.append(start_seed) - p.all_prompts.append(mask_prompt) - p.infotexts.append(updated_info) + try: + p.all_seeds.append(start_seed) + p.all_prompts.append(mask_prompt) + p.infotexts.append(updated_info) + except Exception as e: + pass shared.state.current_image = processed.images[0] diff --git a/scripts/eyemask/script_embedded.py b/scripts/eyemask/script_embedded.py index aef4cac..5528e6f 100644 --- a/scripts/eyemask/script_embedded.py +++ b/scripts/eyemask/script_embedded.py @@ -193,15 +193,21 @@ class EyeMasksEmbeddedCore(EyeMasksCore): updated_info = self.update_info(initial_info, em_params) processed.images.append(processed_em.images[0]) - processed.all_seeds.append(start_seed) - processed.all_prompts.append(save_prompt) - processed.infotexts.append(updated_info) + try: + processed.all_seeds.append(start_seed) + processed.all_prompts.append(save_prompt) + processed.infotexts.append(updated_info) + except Exception as e: + pass if em_include_mask and (n == iterations - 1): processed.images.append(mask) - processed.all_seeds.append(start_seed) - processed.all_prompts.append(mask_prompt) - processed.infotexts.append(updated_info) + try: + processed.all_seeds.append(start_seed) + processed.all_prompts.append(mask_prompt) + processed.infotexts.append(updated_info) + except Exception as e: + pass shared.state.current_image = processed_em.images[0]