Fix support for early version of Invoke AI #36

receyuki · Aug 26, 2023 · f272922 · f272922
1 parent 0422e2e
commit f272922
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sd_prompt_reader/image_data_reader.py b/sd_prompt_reader/image_data_reader.py
@@ -210,7 +210,7 @@ def _ed_format(self):
     def _invoke_metadata(self):
         metadata = json.loads(self._info.get("sd-metadata"))
         image = metadata.get("image")
-        prompt = image.get("prompt")
+        prompt = image.get("prompt")[0].get("prompt") if isinstance(image.get("prompt"), list) else image.get("prompt")
         prompt_index = [prompt.rfind("["), prompt.rfind("]")]
 
         # w/ neg