check for text attr and save

2 years ago · 18379dea36
parent edcff9ab8a
commit 18379dea36
1 changed files with 3 additions and 2 deletions
--- a/server.py
+++ b/server.py
@ -235,8 +235,9 @@ class PromptServer():
                if os.path.isfile(file):
                    with Image.open(file) as original_pil:
                        metadata = PngInfo()
-                        for key in original_pil.text:
-                            metadata.add_text(key, original_pil.text[key])
+                        if hasattr(original_pil,'text'):
+                            for key in original_pil.text:
+                                metadata.add_text(key, original_pil.text[key])
                        original_pil = original_pil.convert('RGBA')
                        mask_pil = Image.open(image.file).convert('RGBA')