Merge pull request #18 from ilian6806/develop

Infotexts bug fixed
ilian6806 · Jun 2, 2023 · 7b803a4 · 7b803a4
2 parents 010807b + 0ef75e1
commit 7b803a4
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 12 deletions.
diff --git a/scripts/eyemask/script.py b/scripts/eyemask/script.py
@@ -212,15 +212,21 @@ def execute(self, p,
                 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
@@ -193,15 +193,21 @@ def execute_postprocess(self, p, processed,
                     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]