From 18379dea366e97105cd01685d629df83c995c150 Mon Sep 17 00:00:00 2001 From: Chris Date: Tue, 29 Aug 2023 18:47:17 +1000 Subject: [PATCH] check for text attr and save --- server.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/server.py b/server.py index 1cac3281..0d7d28a0 100644 --- 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')