From 185d97de5227ac448fbbb19d71ae86644770adb0 Mon Sep 17 00:00:00 2001 From: sani haq Date: Mon, 9 Sep 2024 19:28:55 +0530 Subject: [PATCH] fix: parsing model name from invokeai metadata --- sd_prompt_reader/format/invokeai.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/sd_prompt_reader/format/invokeai.py b/sd_prompt_reader/format/invokeai.py index a2a6284..c687ec6 100644 --- a/sd_prompt_reader/format/invokeai.py +++ b/sd_prompt_reader/format/invokeai.py @@ -64,12 +64,12 @@ def _invoke_metadata(self): self._parameter[p] = remove_quotes( str( ( - data_json.get("model").get("model_name"), - data_json.get("refiner_model").get("model_name"), + data_json.get("model").get("name"), + data_json.get("refiner_model").get("name"), ) ) if has_refiner - else str(data_json.get("model").get("model_name")) + else str(data_json.get("model").get("name")) ) case "seed": self._parameter["seed"] = str(data_json.get("seed"))